Skip to content

Commit cd0fd20

Browse files
committed
chore: 🤖 fix pylint for v1.0.2
1 parent 11016cd commit cd0fd20

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/basicts/runners/basicts_runner.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -890,11 +890,11 @@ def _save_metrics(self) -> None:
890890
"""Save metrics results to self.ckpt_save_dir/test_metrics.json"""
891891

892892
metrics_results = {}
893-
metrics_results["overall"] = {k: self.meter_pool.get_value(f"test/{k}") for k in self.metrics.keys()}
893+
metrics_results["overall"] = {k: self.meter_pool.get_value(f"test/{k}") for k in self.metrics}
894894
if self.eval_horizons is not None:
895895
for i in self.eval_horizons:
896896
metrics_results[f"horizon_{i+1}"] = {
897-
k: self.meter_pool.get_value(f"test/{k}@h{i+1}") for k in self.metrics.keys()}
897+
k: self.meter_pool.get_value(f"test/{k}@h{i+1}") for k in self.metrics}
898898
with open(os.path.join(self.ckpt_save_dir, "test_metrics.json"), "w") as f:
899899
json.dump(metrics_results, f, indent=4)
900900
metrics_path = os.path.join(self.ckpt_save_dir, "test_metrics.json")

0 commit comments

Comments
 (0)