TypeError: __init__() takes 2 positional arguments but 3 were given 1

[NIEYALI]

Hi
I tried to run it, but met problems following: could you help figure out what's wrong?

---

TypeError Traceback (most recent call last)
in ()
59 data_label = [s[1] for s in dataset.samples]
60
---> 61 ss = StratifiedSampler(torch.FloatTensor(data_label), test_size)
62 pre_train_indices, test_indices = ss.gen_sample_array()
63 # The "pre" is necessary to use array to identify train/ val indices with indices generated by second sampler

TypeError: init() takes 2 positional arguments but 3 were given
1
​