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 6, 2024
1 parent ffe9b15 commit 79de1c4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pvnet/models/base_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -621,7 +621,7 @@ def _log_validation_results(self, batch, y_hat, accum_batch_num):
y = y.detach().cpu().numpy()
y_hat = y_hat.detach().cpu().numpy()
time_utc_key = BatchKey[f"{self._target_key_name}_time_utc"]
time_utc = batch[time_utc_key][:, -self.forecast_len:].detach().cpu().numpy()
time_utc = batch[time_utc_key][:, -self.forecast_len :].detach().cpu().numpy()
id_key = BatchKey[f"{self._target_key_name}_id"]
target_id = batch[id_key].detach().cpu().numpy()

Expand Down

0 comments on commit 79de1c4

Please sign in to comment.