tidy up

openclimatefix · Sep 5, 2024 · a17ad48 · a17ad48
1 parent 202385a
commit a17ad48
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pvnet/models/base_model.py b/pvnet/models/base_model.py
@@ -623,10 +623,10 @@ def _log_validation_results(self, batch, y_hat, accum_batch_num):
             y_i = y[i].detach().cpu().numpy()
             y_hat_i = y_hat[i].detach().cpu().numpy()
 
-            time_utc_key = getattr(BatchKey, f"{self._target_key}_time_utc")
+            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 = getattr(BatchKey, f"{self._target_key}_id")
+            id_key = BatchKey[f"{self._target_key}_id"]
             target_id = batch[id_key][i].detach().cpu().numpy()
 
             results_df = pd.DataFrame(