Generator keyword error

[KyriaAnnwyn]

I get this error when try to use safedataloader

dl = nc.SafeDataLoader(ds, batch_size = batch_ss, shuffle = shuffle)
File "/lib/python3.8/site-packages/nonechucks/dataloader.py", line 25, in call
obj = type.call(cls, args, **kwargs)
File "/lib/python3.8/site-packages/nonechucks/dataloader.py", line 138, in init
super(SafeDataLoader, self).init(dataset, **kwargs)
File "/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 277, in init
sampler = RandomSampler(dataset, generator=generator) # type: ignore[arg-type]
TypeError: safe_sampler_callable() got an unexpected keyword argument 'generator'

How can I fix it?

P.S. I'm using PyTorch (1.11.0+cu113)

[alexmehta]

I have the same issue. Did you find any solution?