Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Sep 5, 2024
1 parent 42ed3e7 commit 0444b70
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion pvnet/models/base_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()

Expand Down

0 comments on commit 0444b70

Please sign in to comment.