fix test

EthoML · Nov 15, 2024 · d322bc3 · d322bc3
1 parent 0c10a87
commit d322bc3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/src/vame/__init__.py b/src/vame/__init__.py
@@ -17,3 +17,4 @@
 from vame.util.align_egocentrical import egocentric_alignment
 from vame.util import model_util
 from vame.util import auxiliary
+from vame.util.report import report
diff --git a/tests/test_analysis.py b/tests/test_analysis.py
@@ -312,14 +312,11 @@ def test_report(
     setup_project_and_train_model,
     segmentation_algorithm,
 ):
-    from vame.util.report import report
-
-    report(
+    vame.report(
         config=setup_project_and_train_model["config_path"],
         segmentation_algorithm=segmentation_algorithm,
     )
-
-    reports_path = Path(setup_project_and_train_model["project_path"]) / "reports"
+    reports_path = Path(setup_project_and_train_model["config_data"]["project_path"]) / "reports"
     assert len(list(reports_path.glob("*.png"))) > 0