Fix VP test issue (#3482)

Fix VP issue
openvinotoolkit · May 10, 2024 · 94d0da6 · 94d0da6
1 parent eaedc34
commit 94d0da6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/e2e/cli/test_cli.py b/tests/e2e/cli/test_cli.py
@@ -67,7 +67,7 @@ def test_otx_e2e_cli(
         "--engine.device",
         fxt_accelerator,
         "--max_epochs",
-        "1" if task in ("zero_shot_visual_prompting") else "2",
+        "1" if task in (OTXTaskType.ZERO_SHOT_VISUAL_PROMPTING) else "2",
         *fxt_cli_override_command_per_task[task],
     ]