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 Feb 22, 2023
1 parent 7804ea6 commit 266e3b3
Showing 1 changed file with 2 additions and 7 deletions.
9 changes: 2 additions & 7 deletions elk/evaluation/evaluate.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,16 +29,11 @@ def evaluate_reporters(cfg: EvaluateConfig):

_, hiddens = normalize(hiddens, hiddens, cfg.normalization)

reporter_root_path = (
elk_cache_dir() / cfg.source / "reporters"
)
reporter_root_path = elk_cache_dir() / cfg.source / "reporters"

reporters = []
for path in reporter_root_path.glob("*.pt"):
reporter = torch.load(
path,
map_location=cfg.device
)
reporter = torch.load(path, map_location=cfg.device)
reporters.append(reporter)

transfer_eval = reporter_root_path / "transfer_eval"
Expand Down

0 comments on commit 266e3b3

Please sign in to comment.