Skip to content

Commit

Permalink
fix test
Browse files Browse the repository at this point in the history
  • Loading branch information
azhou-determined committed Oct 16, 2024
1 parent bc1c61c commit 4bb779b
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions harness/tests/experiment/pytorch/test_pytorch_trial.py
Original file line number Diff line number Diff line change
Expand Up @@ -1469,10 +1469,9 @@ def run_identity(tmp_path: pathlib.Path):
checkpoint_dir = str(tmp_path.joinpath("checkpoint"))

config = utils.load_config(utils.fixtures_path("pytorch_identity/distributed.yaml"))
hparams = config["hyperparameters"]

exp_config = utils.make_default_exp_config(
hparams,
hparams={},
scheduling_unit=1,
searcher_metric="validation_loss",
checkpoint_dir=checkpoint_dir,
Expand All @@ -1489,7 +1488,7 @@ def run_identity(tmp_path: pathlib.Path):

trial, trial_controller = pytorch_utils.create_trial_and_trial_controller(
trial_class=trial_class,
hparams=hparams,
hparams={},
slots_per_trial=2,
max_batches=16,
min_validation_batches=1,
Expand Down

0 comments on commit 4bb779b

Please sign in to comment.