search

p-lambda / wilds

A machine learning benchmark of in-the-wild distribution shifts, with data loaders, evaluators, and default models.
https://wilds.stanford.edu
MIT License
551 stars 128 forks source link

WILDSSubset __getitem__ transform is broken #77

Closed lavoiems closed 3 years ago

lavoiems commented 3 years ago

https://github.com/p-lambda/wilds/blob/aa434b540fdbfb3e6d19a0857569abd52bcd44b3/wilds/datasets/wilds_dataset.py#L456

self.transform is given 2 inputs. But, from the README, transform can be defined as a torchvision.transforms which takes 1 input. The result is the following error:

TypeError: __call__() takes 2 positional arguments but 3 were given
kohpangwei commented 3 years ago

Hi Sam,

Thank you for reporting this! We'll work on fixing it and will keep you posted.

kohpangwei commented 3 years ago

This should be fixed in the latest release. Thank you for reporting it!