Unable to use FewShotSampler

[SeiunSky0131]

For example, I am supposed to construct a sampler using

sampler10 = FewShotSampler(num_examples_per_label = 5, also_sample_dev = True, num_examples_per_label_dev = 5)

I should then feed the data

sampler10(train_dataset = train_dataset, valid_dataset = valid_dataset, seed = 42)

where

train_data = DataLoader(train_df, batch_size = 32, shuffle = True)
train_dataset = train_data.dataset

valid_data = DataLoader(dev_df, batch_size = 32, shuffle = True)
valid_dataset = valid_data.dataset

But it give errors

KeyError                                  Traceback (most recent call last)
[/tmp/ipykernel_28975/1100549245.py](https://file+.vscode-resource.vscode-cdn.net/tmp/ipykernel_28975/1100549245.py) in 
     13 sampler500 = FewShotSampler(num_examples_per_label = 250, also_sample_dev = True, num_examples_per_label_dev = 250)
     14 
---> 15 sampler10(train_dataset = train_dataset, valid_dataset = valid_dataset, seed = 42)
     16 
...
-> 3623                 raise KeyError(key) from err
   3624             except TypeError:
   3625                 # If we have a listlike key, _check_indexing_error will raise

KeyError: 0