fixed tests

mmcdermott · Sep 9, 2024 · c631e93 · c631e93
1 parent 8d12aed
commit c631e93
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git 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
@@ -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,