diff --git a/tests/fixtures/__init__.py b/tests/fixtures/__init__.py index add8cbad..373b8079 100644 --- a/tests/fixtures/__init__.py +++ b/tests/fixtures/__init__.py @@ -24,5 +24,11 @@ unet_3d_distance_run, hot_distance_run, ) -from .tasks import dummy_task, distance_task, onehot_task, six_onehot_task, hot_distance_task +from .tasks import ( + dummy_task, + distance_task, + onehot_task, + six_onehot_task, + hot_distance_task, +) from .trainers import dummy_trainer, gunpowder_trainer diff --git a/tests/fixtures/runs.py b/tests/fixtures/runs.py index d66323bc..c842db11 100644 --- a/tests/fixtures/runs.py +++ b/tests/fixtures/runs.py @@ -20,6 +20,7 @@ def distance_run( num_iterations=10, ) + @pytest.fixture() def hot_distance_run( six_class_datasplit, @@ -38,7 +39,6 @@ def hot_distance_run( ) - @pytest.fixture() def dummy_run( dummy_datasplit, diff --git a/tests/fixtures/tasks.py b/tests/fixtures/tasks.py index 2dcffc52..bd8b2508 100644 --- a/tests/fixtures/tasks.py +++ b/tests/fixtures/tasks.py @@ -28,6 +28,7 @@ def distance_task(): tol_distance=10, ) + @pytest.fixture() def hot_distance_task(): yield HotDistanceTaskConfig( @@ -44,6 +45,7 @@ def hot_distance_task(): tol_distance=10, ) + @pytest.fixture() def onehot_task(): yield OneHotTaskConfig(