Merge pull request #15 from openclimatefix/jack/python39

Must use Python 3.9 or above

Author: jacobbieker

README.md

@@ -14,6 +14,8 @@ pip install -e .
 
 Alternatively, you can also install a usually older version through ```pip install metnet```
 
+Please ensure that you're using Python version 3.9 or above.
+
 ## Data
 
 While the exact training data used for both MetNet and MetNet-2 haven't been released, the papers do go into some detail as to the inputs, which were GOES-16 and MRMS precipitation data, as well as the time period covered. We will be making those splits available, as well as a larger dataset that covers a longer time period, with [HuggingFace Datasets](https://huggingface.co/datasets/openclimatefix/goes-mrms)!

tests/test_model.py

@@ -28,6 +28,7 @@ def test_metnet_creation():
     )
     assert not torch.isnan(out).any(), "Output included NaNs"
 
+
 def test_metnet_backwards():
     model = MetNet(
         hidden_dim=32,
@@ -36,7 +37,7 @@ def test_metnet_backwards():
         output_channels=12,
         sat_channels=12,
         input_size=32,
-        )
+    )
     # MetNet expects original HxW to be 4x the input size
     x = torch.randn((2, 12, 16, 128, 128))
     out = model(x)
@@ -47,7 +48,7 @@ def test_metnet_backwards():
         12,
         8,
         8,
-        )
+    )
     y = torch.randn((2, 24, 12, 8, 8))
     F.mse_loss(out, y).backward()
     assert not torch.isnan(out).any(), "Output included NaNs"
@@ -86,6 +87,7 @@ def test_metnet2_creation():
     )
     assert not torch.isnan(out).any(), "Output included NaNs"
 
+
 def test_metnet2_backward():
     model = MetNet2(
         forecast_steps=8,
@@ -95,7 +97,7 @@ def test_metnet2_backward():
         lstm_channels=32,
         encoder_channels=64,
         center_crop_size=16,
-        )
+    )
     # MetNet expects original HxW to be 4x the input size
     x = torch.randn((2, 6, 12, 256, 256))
     out = model(x)
@@ -106,7 +108,7 @@ def test_metnet2_backward():
         12,
         64,
         64,
-        )
-    y = torch.rand((2,8,12,64,64))
+    )
+    y = torch.rand((2, 8, 12, 64, 64))
     F.mse_loss(out, y).backward()
-    assert not torch.isnan(out).any(), "Output included NaNs"
+    assert not torch.isnan(out).any(), "Output included NaNs"