update to pvnet v2.4

openclimatefix · Nov 22, 2023 · cf0dbb0 · cf0dbb0
1 parent 800667c
commit cf0dbb0
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/pvnet_summation/models/base_model.py b/pvnet_summation/models/base_model.py
@@ -267,4 +267,4 @@ def configure_optimizers(self):
         if self.lr is not None:
             # Use learning rate found by learning rate finder callback
             self._optimizer.lr = self.lr
-        return self._optimizer(self.parameters())
+        return self._optimizer(self)
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -106,9 +106,10 @@ def sample_batch(sample_datamodule):
 
 @pytest.fixture()
 def model_kwargs():
+    # These kwargs define the pvnet model which the summation model uses
     kwargs = dict(
         model_name="openclimatefix/pvnet_v2",
-        model_version="898630f3f8cd4e8506525d813dd61c6d8de86144",
+        model_version="805ca9b2ee3120592b0b70b7c75a454e2b4e4bec",
     )
     return kwargs
 

diff --git a/tests/data/sample_batches/train/000000.pt b/tests/data/sample_batches/train/000000.pt