diff --git a/tests/test_configs.py b/tests/test_configs.py index 5ad1a13..d0bbc45 100644 --- a/tests/test_configs.py +++ b/tests/test_configs.py @@ -65,7 +65,6 @@ def test_model_config(model_launcher_override, imputer, normalization, tmp_path) "++tabularization.filtered_code_metadata_fp": code_metadata_fp, "++tabularization.min_code_inclusion_count": "0", "task_name": "foo_bar", - "input_label_dir": "/qux/", } pl.DataFrame({"code": ["E", "D", "A"], "count": [4, 3, 2]}).write_parquet(code_metadata_fp) diff --git a/tests/test_tabularize.py b/tests/test_tabularize.py index d8a005f..006252d 100644 --- a/tests/test_tabularize.py +++ b/tests/test_tabularize.py @@ -351,11 +351,11 @@ def test_tabularize(tmp_path): expected_output_dir = Path(cfg.output_model_dir) output_files = list(expected_output_dir.glob("**/*.json")) assert len(output_files) == 2 - shutil.rmtree(expected_output_dir) log_dir = Path(cfg.path.model_log_dir) - log_csv = list(log_dir.glob("**/*.log")) - assert len(log_csv) == 2 + log_files = list(log_dir.glob("**/*.log")) + assert len(log_files) == 1 + shutil.rmtree(expected_output_dir) sklearnmodel_config = { **shared_config,