Correcting KeyError happening in quickstart notebook when "level" is scalar

pangeo-data / WeatherBench

A benchmark dataset for data-driven weather forecasting

MIT License

694 stars 166 forks source link

Correcting KeyError happening in quickstart notebook when "level" is scalar #44

Closed Deathn0t closed 2 years ago

Deathn0t commented 2 years ago

Hello, when pulling the repo and running the quickstart notebook I got KeyError with the load_test_data and DataGenerator because level is a scalar value. I proposed a fix which I think will not impact other utilisation of this code (when level is not a scalar).

raspstephan commented 2 years ago

Thanks for that. Happy to merge.