Merge remote-tracking branch 'origin/jacob/windnet' into jacob/windnet

# Conflicts:
#	pvnet/models/base_model.py

Author: jacobbieker

pvnet/models/utils.py

@@ -99,9 +99,12 @@ def __bool__(self):
 
     # @staticmethod
     def _filter_batch_dict(self, d):
-        keep_keys = (
-            [BatchKey[self.key_to_keep], BatchKey[f"{self.key_to_keep}_id"], BatchKey[f"{self.key_to_keep}_t0_idx"], BatchKey[f"{self.key_to_keep}_time_utc"]]
-        )
+        keep_keys = [
+            BatchKey[self.key_to_keep],
+            BatchKey[f"{self.key_to_keep}_id"],
+            BatchKey[f"{self.key_to_keep}_t0_idx"],
+            BatchKey[f"{self.key_to_keep}_time_utc"],
+        ]
         return {k: v for k, v in d.items() if k in keep_keys}
 
     def append(self, batch: dict[BatchKey, list[torch.Tensor]]):

pvnet/utils.py

@@ -255,7 +255,7 @@ def _get_numpy(key):
     y = batch[y_key][:, :, 0].cpu().numpy()  # Select the one it is trained on
     y_hat = y_hat.cpu().numpy()
     gsp_ids = batch[y_id_key][:, 0].cpu().numpy().squeeze()
-    t0_idx = int(batch[t0_idx_key])
+    int(batch[t0_idx_key])
     plotting_name = key_to_plot.upper()
 
     gsp_ids = batch[y_id_key].cpu().numpy().squeeze()