PVNet 5.0.8__py3-none-any.whl → 5.0.9__py3-none-any.whl

pvnet/training/lightning_module.py

@@ -105,8 +105,9 @@ class PVNetLightningModule(pl.LightningModule):
         """Run training step"""
         y_hat = self.model(batch)
 
-        # Batch is adapted in the model forward method, but needs to be adapted here too
-        batch = self.model._adapt_batch(batch)
+        # Batch may be adapted in the model forward method, would need adapting here too
+        if self.model.adapt_batches:
+            batch = self.model._adapt_batch(batch)
 
         y = batch[self.model._target_key][:, -self.model.forecast_len :]
 
@@ -211,8 +212,9 @@ class PVNetLightningModule(pl.LightningModule):
             with torch.no_grad():
                 y_hat = self.model(batch)
             
-            # Batch is adapted in the model forward method, but needs to be adapted here too
-            batch = self.model._adapt_batch(batch)
+            # Batch may be adapted in the model forward method, would need adapting here too
+            if self.model.adapt_batches:
+                batch = self.model._adapt_batch(batch)
             
             fig = plot_sample_forecasts(
                 batch,
@@ -233,8 +235,9 @@ class PVNetLightningModule(pl.LightningModule):
         """Run validation step"""
 
         y_hat = self.model(batch)
-        # Batch is adapted in the model forward method, but needs to be adapted here too
-        batch = self.model._adapt_batch(batch)
+        # Batch may be adapted in the model forward method, would need adapting here too
+        if self.model.adapt_batches:
+            batch = self.model._adapt_batch(batch)
 
         # Internally store the val predictions
         self._store_val_predictions(batch, y_hat)

{pvnet-5.0.8.dist-info → pvnet-5.0.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PVNet
-Version: 5.0.8
+Version: 5.0.9
 Summary: PVNet
 Author-email: Peter Dudfield <info@openclimatefix.org>
 Requires-Python: >=3.11

{pvnet-5.0.8.dist-info → pvnet-5.0.9.dist-info}/RECORD

@@ -22,11 +22,11 @@ pvnet/models/late_fusion/site_encoders/__init__.py,sha256=QoUiiWWFf12vEpdkw0gO4T
 pvnet/models/late_fusion/site_encoders/basic_blocks.py,sha256=iEB_N7ZL5HMQ1hZM6H32A71GCwP7YbErUx0oQF21PQM,1042
 pvnet/models/late_fusion/site_encoders/encoders.py,sha256=k4z690cfcP6J4pm2KtDujHN-W3uOl7QY0WvBIu1tM8c,11703
 pvnet/training/__init__.py,sha256=FKxmPZ59Vuj5_mXomN4saJ3En5M-aDMxSs6OttTQOcg,49
-pvnet/training/lightning_module.py,sha256=B8fDpWIcM7v-653cbAtoNM4qQ9wB88IqWeRn0YIK2AU,13115
+pvnet/training/lightning_module.py,sha256=GVqdi5ALFo9-_WRYeyMMj2qH_k4gPxQ2sG6FhL_wRFE,13242
 pvnet/training/plots.py,sha256=4xID7TBA4IazaARaCN5AoG5fFPJF1wIprn0y6I0C31c,2469
 pvnet/training/train.py,sha256=zj9JMi9C6W68vGsQUBapWkJ4aDzDuJFMv0IVjO73s1k,5215
-pvnet-5.0.8.dist-info/licenses/LICENSE,sha256=tKUnlSmcLBWMJWkHx3UjZGdrjs9LidGwLo0jsBUBAwU,1077
-pvnet-5.0.8.dist-info/METADATA,sha256=LUBs04lvJcUzUzM--aswF_lyBwQ9RUB-cJLuhg44j8g,18043
-pvnet-5.0.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pvnet-5.0.8.dist-info/top_level.txt,sha256=4mg6WjeW05SR7pg3-Q4JRE2yAoutHYpspOsiUzYVNv0,6
-pvnet-5.0.8.dist-info/RECORD,,
+pvnet-5.0.9.dist-info/licenses/LICENSE,sha256=tKUnlSmcLBWMJWkHx3UjZGdrjs9LidGwLo0jsBUBAwU,1077
+pvnet-5.0.9.dist-info/METADATA,sha256=Z909YP0GU68upJxamWI1BlHI4vZjbSUAAlEnIt2V_fc,18043
+pvnet-5.0.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pvnet-5.0.9.dist-info/top_level.txt,sha256=4mg6WjeW05SR7pg3-Q4JRE2yAoutHYpspOsiUzYVNv0,6
+pvnet-5.0.9.dist-info/RECORD,,