try and except around odd error

openclimatefix · Sep 5, 2024 · 42ed3e7 · 42ed3e7
1 parent e227b65
commit 42ed3e7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pvnet/models/base_model.py b/pvnet/models/base_model.py
@@ -624,8 +624,11 @@ def _log_validation_results(self, batch, y_hat, accum_batch_num):
             y_hat_i = y_hat[i].detach().cpu().numpy()
 
             print(BatchKey._member_map_)
+            try:
+                time_utc_key = BatchKey[f"{self._target_key}_time_utc"]
+            except Exception as e:
+                raise Exception(f"Failed to find time_utc key for {self._target_key}, {BatchKey._member_map_}, {e}")
 
-            time_utc_key = BatchKey[f"{self._target_key}_time_utc"]
             time_utc = batch[time_utc_key][i, -self.forecast_len :].detach().cpu().numpy()
 
             id_key = BatchKey[f"{self._target_key}_id"]