diff --git a/src/vame/__init__.py b/src/vame/__init__.py index e61bbf29..4156c778 100644 --- 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 index 9b8a54ac..167cb0e6 100644 --- 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