diff --git a/tests/e2e/cli/visual_prompting/test_visual_prompting.py b/tests/e2e/cli/visual_prompting/test_visual_prompting.py index 43fa07a9f1b..2749a09f347 100644 --- a/tests/e2e/cli/visual_prompting/test_visual_prompting.py +++ b/tests/e2e/cli/visual_prompting/test_visual_prompting.py @@ -60,11 +60,13 @@ templates_ids = [template.model_template_id + f"-{i+1}" for i, template in enumerate(templates)] else: - templates = ( - Registry("src/otx/algorithms/visual_prompting", experimental=True) + templates = [ + template + for template in Registry("src/otx/algorithms/visual_prompting", experimental=True) .filter(task_type="VISUAL_PROMPTING") .templates - ) + if "Zero_Shot" not in template.name + ] templates_ids = [template.model_template_id for template in templates]