Unable to load sustainbench dataloaders

[yenniejun]

Hi, thanks for sharing everything! Was wondering if you could provide the code for loading the dataloaders for loading the datasets as PyTorch tensors? I could not find the details on the website as instructed on the ReadMe.

I cloned the repo and imported sustainbench (from sustainbench import sustainbench). Then, I tried to run sustainbench.get_dataset('poverty') but am running into import issues such as module 'sustainbench' has no attribute 'supported_datasets'. I'm wondering if this process would be easier with some extra documentation or sample code for getting the dataloaders. Thanks so much!

@chrisyeh96