Fix name

openclimatefix · Dec 11, 2023 · 88ff9df · 88ff9df
1 parent 4bbf653
commit 88ff9df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pvnet/utils.py b/pvnet/utils.py
@@ -252,7 +252,7 @@ def _get_numpy(key):
     time_utc_key = BatchKey.gsp_time_utc if key_to_plot == "gsp" else BatchKey.sensor_time_utc
     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].cpu().numpy().squeeze()
+    gsp_ids = batch[y_id_key][:,0].cpu().numpy().squeeze()
     t0_idx = int(batch[t0_idx_key])
     plotting_name = "GSP" if key_to_plot == "gsp" else "Sensor"