Added NNCF steps

Author: pfinashx

external/anomaly/ote_anomalib/nncf_task.py

@@ -36,6 +36,7 @@
 from ote_sdk.entities.datasets import DatasetEntity
 from ote_sdk.entities.model import (
     ModelEntity,
+    ModelFormat,
     ModelOptimizationType,
     ModelPrecision,
     OptimizationMethod,
@@ -183,6 +184,8 @@ def optimize(
         self.trainer = Trainer(**self.config.trainer, logger=False, callbacks=callbacks)
         self.trainer.fit(model=self.model, datamodule=datamodule)
         self.compression_ctrl = nncf_callback.nncf_ctrl
+        output_model.model_format = ModelFormat.BASE_FRAMEWORK
+        output_model.optimization_type = ModelOptimizationType.NN
         self.save_model(output_model)
 
         logger.info("Training completed.")

external/anomaly/tests/test_ote_training.py

@@ -409,7 +409,7 @@ def test(self,
              test_parameters,
              test_case_fx, data_collector_fx,
              cur_test_expected_metrics_callback_fx):
-        if "nncf" in test_parameters["test_stage"]:
-            pytest.xfail("NNCF not yet supported for Anomaly Classification")
+        if "nncf_graph" in test_parameters["test_stage"]:
+            pytest.xfail("The models has no a reference NNCF graph yet")
         test_case_fx.run_stage(test_parameters['test_stage'], data_collector_fx,
                                cur_test_expected_metrics_callback_fx)