PVNet 5.0.21__tar.gz → 5.0.23__tar.gz

{pvnet-5.0.21 → pvnet-5.0.23}/PKG-INFO

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

{pvnet-5.0.21 → pvnet-5.0.23}/PVNet.egg-info/PKG-INFO

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

{pvnet-5.0.21 → pvnet-5.0.23}/pvnet/training/lightning_module.py

@@ -38,6 +38,7 @@ class PVNetLightningModule(pl.LightningModule):
 
         self.model = model
         self._optimizer = optimizer
+        self.save_all_validation_results = save_all_validation_results
 
         # Model must have lr to allow tuning
         # This setting is only used when lr is tuned with callback
@@ -151,13 +152,13 @@ class PVNetLightningModule(pl.LightningModule):
     def _store_val_predictions(self, batch: TensorBatch, y_hat: torch.Tensor) -> None:
         """Internally store the validation predictions"""
         
-        taregt_key = self.model._target_key
+        target_key = self.model._target_key
 
-        y = batch[taregt_key][:, -self.model.forecast_len :].cpu().numpy()
+        y = batch[target_key][:, -self.model.forecast_len :].cpu().numpy()
         y_hat = y_hat.cpu().numpy() 
-        ids = batch[f"{taregt_key}_id"].cpu().numpy()
+        ids = batch[f"{target_key}_id"].cpu().numpy()
         init_times_utc = pd.to_datetime(
-            batch[f"{taregt_key}_time_utc"][:, self.model.history_len+1]
+            batch[f"{target_key}_time_utc"][:, self.model.history_len+1]
             .cpu().numpy().astype("datetime64[ns]")
         )
 
@@ -312,7 +313,7 @@ class PVNetLightningModule(pl.LightningModule):
             self.log_dict(extreme_error_metrics, on_step=False, on_epoch=True)
 
             # Optionally save all validation results - these are overridden each epoch
-            if self.hparams.save_all_validation_results:
+            if self.save_all_validation_results:
                 # Add attributes
                 ds_val_results.attrs["epoch"] = self.current_epoch