KNN few shot fixes

[adam-simple]

• Don't pass valset to bootstrap fewshot
• forward_pass unpacks appropriately
• few shot bootstrap args to pass through during forward pass

[adam-simple]

It might make sense to put a warning in the KNN call if **kwargs is empty. this can easily happen if someone calls .forward without keyword arguments, instead using args without keywords, like module.forward(input) or something. This currently would let KNN silently vectorize an empty list which will return some constant vector and so it will always return the same k likely irrelevant vectors.

[okhat]

Thanks! Probably remove *args altogether then?

[okhat]

tagging @adam-simple

[adam-simple]

done

[arnavsinghvi11]

Thanks @adam-simple !