diff --git a/pvnet/models/base_model.py b/pvnet/models/base_model.py
index f0b6e813..7f9f54f0 100644
--- a/pvnet/models/base_model.py
+++ b/pvnet/models/base_model.py
@@ -627,7 +627,9 @@ def _log_validation_results(self, batch, y_hat, accum_batch_num):
             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}")
+                raise Exception(
+                    f"Failed to find time_utc key for {self._target_key}, {BatchKey._member_map_}, {e}"
+                )
 
             time_utc = batch[time_utc_key][i, -self.forecast_len :].detach().cpu().numpy()