From c4a11d5d6c97e9e273f5c02977de16eb36e909f2 Mon Sep 17 00:00:00 2001 From: Haroon <106879583+sky0walker99@users.noreply.github.com> Date: Mon, 2 Dec 2024 23:41:35 +0530 Subject: [PATCH] Resolved Unecessary modification Signed-off-by: Haroon <106879583+sky0walker99@users.noreply.github.com> --- src/anomalib/deploy/export.py | 2 -- src/anomalib/engine/engine.py | 6 ------ 2 files changed, 8 deletions(-) diff --git a/src/anomalib/deploy/export.py b/src/anomalib/deploy/export.py index 80de3733e8..aae359c035 100644 --- a/src/anomalib/deploy/export.py +++ b/src/anomalib/deploy/export.py @@ -58,8 +58,6 @@ class CompressionType(str, Enum): INT8 = "int8" INT8_PTQ = "int8_ptq" INT8_ACQ = "int8_acq" - POT = "pot" - NNCF = "nncf" class InferenceModel(nn.Module): diff --git a/src/anomalib/engine/engine.py b/src/anomalib/engine/engine.py index 0817f658d0..b537819729 100644 --- a/src/anomalib/engine/engine.py +++ b/src/anomalib/engine/engine.py @@ -915,7 +915,6 @@ def export( Raises: ValueError: If Dataset, Datamodule, and transform are not provided. - ValueError: If an unsupported compression type is specified for non-OpenVINO export types. TypeError: If path to the transform file is not a string or Path. CLI Usage: @@ -976,11 +975,6 @@ def export( else: logging.error(f"Export type {export_type} is not supported yet.") - if compression_type and export_type != ExportType.OPENVINO: - msg = f"Compression type {compression_type} is only applicable for OpenVIVO Export Type." - logging.error(f"Compression type {compression_type} is only applicable for OpenVIVO Export Type.") - raise ValueError(msg) - if exported_model_path: logging.info(f"Exported model to {exported_model_path}") return exported_model_path