From 0444b700fa5126ecc223d63c441742e87c1977dd Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 5 Sep 2024 16:11:52 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- pvnet/models/base_model.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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()