Fix string

openclimatefix · Dec 11, 2023 · b1b43c0 · b1b43c0
1 parent 6056bab
commit b1b43c0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pvnet/utils.py b/pvnet/utils.py
@@ -254,7 +254,7 @@ def _get_numpy(key):
     y = batch[y_key].cpu().numpy()
     y_hat = y_hat.cpu().numpy()
     gsp_ids = batch[y_id_key].cpu().numpy().squeeze()
-    t0_idx = batch[t0_idx_key]
+    t0_idx = int(batch[t0_idx_key])
     plotting_name = "GSP" if key_to_plot == "gsp" else "Sensor"
 
     times_utc = batch[time_utc_key].cpu().numpy().squeeze().astype("datetime64[s]")