Applying comments v.1

openvinotoolkit · Feb 22, 2022 · f58b4bd · f58b4bd
1 parent 4c1f39e
commit f58b4bd
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 122 deletions.
diff --git a/external/anomaly/tests/config.py b/external/anomaly/tests/config.py
diff --git a/external/anomaly/tests/e2e_test_system.py b/external/anomaly/tests/e2e_test_system.py
diff --git a/external/anomaly/tests/test_ote_api.py b/external/anomaly/tests/test_ote_api.py
diff --git a/external/anomaly/tests/test_ote_training.py b/external/anomaly/tests/test_ote_training.py
@@ -55,6 +55,7 @@
                                                        OTETestPotEvaluationAction,
                                                        OTETestNNCFAction,
                                                        OTETestNNCFEvaluationAction,
+                                                       OTETestNNCFGraphAction,
                                                        OTETestNNCFExportAction,
                                                        OTETestNNCFExportEvaluationAction)
 
@@ -112,6 +113,7 @@ def get_anomaly_test_action_classes() -> List[Type[BaseOTETestAction]]:
         OTETestPotEvaluationAction,
         OTETestNNCFAction,
         OTETestNNCFEvaluationAction,
+        OTETestNNCFGraphAction,
         OTETestNNCFExportAction,
         OTETestNNCFExportEvaluationAction,
     ]
@@ -130,7 +132,7 @@ def test_bunches(self) -> List[Dict[str, Any]]:
                        'ote_anomaly_classification_padim',
                        'ote_anomaly_classification_stfpm',
                     ],
-                    dataset_name='mvtec',
+                    dataset_name='mvtec_short',
                     usecase='precommit',
                 ),
                 dict(