update 'tests/unit/launchers'

tdspora · Nov 14, 2024 · 9624d52 · 9624d52
1 parent 5ebb795
commit 9624d52
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/tests/unit/launchers/test_launch_infer.py b/src/tests/unit/launchers/test_launch_infer.py
@@ -208,8 +208,8 @@ def test_infer_table_with_valid_parameter_reports(
 @pytest.mark.parametrize(
     "first_value, second_value",
     [
-        (pv, i) for pv in ["accuracy", "metrics_only"]
-        for i in ["accuracy", "metrics_only"]
+        (pv, i) for pv in INFER_REPORT_TYPES
+        for i in INFER_REPORT_TYPES
     ]
 )
 @patch.object(Worker, "launch_infer")

diff --git a/src/tests/unit/launchers/test_launch_train.py b/src/tests/unit/launchers/test_launch_train.py
@@ -291,8 +291,8 @@ def test_train_table_with_valid_parameter_reports(
 @pytest.mark.parametrize(
     "first_value, second_value",
     [
-        (pv, i) for pv in ["accuracy", "metrics_only", "sample"]
-        for i in ["accuracy", "metrics_only", "sample"]
+        (pv, i) for pv in TRAIN_REPORT_TYPES
+        for i in TRAIN_REPORT_TYPES
     ]
 )
 @patch.object(Worker, "launch_train")