Skip to content

Commit

Permalink
fix a setting in test_evaluate_model
Browse files Browse the repository at this point in the history
  • Loading branch information
OuyangWenyu committed Nov 29, 2023
1 parent 74bc661 commit 3ae4016
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions tests/test_evaluate_model.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
"""
Author: Wenyu Ouyang
Date: 2023-09-18 14:34:53
LastEditTime: 2023-10-03 19:44:48
LastEditTime: 2023-11-29 17:55:05
LastEditors: Wenyu Ouyang
Description: A simple evaluate model test
FilePath: \torchhydro\tests\test_evaluate_model.py
Expand All @@ -20,7 +20,7 @@

@pytest.fixture()
def config_data():
project_name = "test_camels/exp2"
project_name = "test_camels/exp1"
weight_dir = os.path.join(
os.getcwd(),
"results",
Expand Down Expand Up @@ -67,6 +67,7 @@ def config_data():
opt="Adadelta",
which_first_tensor="sequence",
weight_path=weight_path,
continue_train=False,
)
config_data = default_config_file()
update_cfg(config_data, args)
Expand Down

0 comments on commit 3ae4016

Please sign in to comment.