search

openclimatefix / metnet

PyTorch Implementation of Google Research's MetNet and MetNet-2
MIT License
231 stars 47 forks source link

A bug in the ConvLSTMCell #14

Closed Hinode closed 2 years ago

Hinode commented 2 years ago

Describe the bug When I test the "test_model.py", I got the following error message:

metnet\layers\ConvLSTM.py", line 54, in ConvLSTMCell def forward(self, x: torch.Tensor, prev_state: list) -> tuple[torch.Tensor, torch.Tensor]: TypeError: 'type' object is not subscriptable

JackKelly commented 2 years ago

Please ensure you're using Python version 3.9 or above.

Sorry this isn't documented. I'll add something to the README now

(Python 3.9 introduced the ability to use tuple as a type hint)

Hinode commented 2 years ago

Thank you, Jack.