Update test

tests/integration/api/test_augmentation.py

@@ -48,10 +48,13 @@ def _test_aug_cls(
     for switches in itertools.product([True, False], repeat=len(configurable_augs)):
         # Configure on/off
         for aug_name, switch in zip(configurable_augs, switches):
+            aug_found = False
             for aug_config in train_config["transforms"]:
                 if aug_name in aug_config["class_path"]:
                     aug_config["enable"] = switch
+                    aug_found = True
                     break
+            assert aug_found, f"{aug_name} not found in {recipe}"
         # Create dataset
         dataset = OTXDatasetFactory.create(
             task=task,
@@ -68,7 +71,9 @@ def _test_aug_cls(
             assert img_shape == data.img_info.img_shape
 
 
-CLS_RECIPES = [recipe for recipe in pytest.RECIPE_LIST if "_cls" in recipe and "semi" not in recipe]
+CLS_RECIPES = [
+    recipe for recipe in pytest.RECIPE_LIST if "_cls" in recipe and "semi" not in recipe and "tv_" not in recipe
+]
 
 
 @pytest.mark.parametrize(
@@ -87,4 +92,4 @@ def test_aug_cls(
         "GaussianBlur",
         "GaussianNoise",
     ]
-    _test_aug_cls(recipe, fxt_target_dataset_per_task, configurable_augs)
+