Skip to content

Commit c4d84ce

Browse files
committed
create rather than read in base_params
1 parent ee2b4ae commit c4d84ce

2 files changed

Lines changed: 2 additions & 33 deletions

File tree

tests/test_parameter_plots.py

Lines changed: 1 addition & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -14,22 +14,7 @@
1414

1515
# Load in test results and parameters
1616
CUR_PATH = os.path.abspath(os.path.dirname(__file__))
17-
if sys.version_info[1] < 11:
18-
base_params = utils.safe_read_pickle(
19-
os.path.join(CUR_PATH, "test_io_data", "model_params_baseline.pkl")
20-
)
21-
elif sys.version_info[1] == 11:
22-
base_params = utils.safe_read_pickle(
23-
os.path.join(
24-
CUR_PATH, "test_io_data", "model_params_baseline_v311.pkl"
25-
)
26-
)
27-
else:
28-
base_params = utils.safe_read_pickle(
29-
os.path.join(
30-
CUR_PATH, "test_io_data", "model_params_baseline_v312.pkl"
31-
)
32-
)
17+
base_params = Specifications()
3318
base_taxfunctions = utils.safe_read_pickle(
3419
os.path.join(CUR_PATH, "test_io_data", "TxFuncEst_baseline.pkl")
3520
)

tests/test_parameter_tables.py

Lines changed: 1 addition & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -11,23 +11,7 @@
1111

1212
# Load in test results and parameters
1313
CUR_PATH = os.path.abspath(os.path.dirname(__file__))
14-
15-
if sys.version_info[1] < 11:
16-
base_params = utils.safe_read_pickle(
17-
os.path.join(CUR_PATH, "test_io_data", "model_params_baseline.pkl")
18-
)
19-
elif sys.version_info[1] == 11:
20-
base_params = utils.safe_read_pickle(
21-
os.path.join(
22-
CUR_PATH, "test_io_data", "model_params_baseline_v311.pkl"
23-
)
24-
)
25-
else:
26-
base_params = utils.safe_read_pickle(
27-
os.path.join(
28-
CUR_PATH, "test_io_data", "model_params_baseline_v312.pkl"
29-
)
30-
)
14+
base_params = Specifications()
3115
base_taxfunctions = utils.safe_read_pickle(
3216
os.path.join(CUR_PATH, "test_io_data", "TxFuncEst_baseline.pkl")
3317
)

0 commit comments

Comments
 (0)