From eb5bbb292f0f215987585d08fd19714c965874e8 Mon Sep 17 00:00:00 2001
From: liord <lior.dikstein@altair-semi.com>
Date: Tue, 10 Dec 2024 12:49:10 +0200
Subject: [PATCH] Fix OperatorSetNames Enum

---
 .../target_platform_capabilities/schema/v1.py          | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/model_compression_toolkit/target_platform_capabilities/schema/v1.py b/model_compression_toolkit/target_platform_capabilities/schema/v1.py
index a2be27f5a..306da315f 100644
--- a/model_compression_toolkit/target_platform_capabilities/schema/v1.py
+++ b/model_compression_toolkit/target_platform_capabilities/schema/v1.py
@@ -23,11 +23,6 @@
     _current_tp_model
 
 class OperatorSetNames(Enum):
-    OPSET_NO_QUANTIZATION = "NoQuantization"
-    OPSET_QUANTIZATION_PRESERVING = "QuantizationPreserving"
-    OPSET_DIMENSION_MANIPULATION_OPS_WITH_WEIGHTS = "DimensionManipulationOpsWithWeights"
-    OPSET_DIMENSION_MANIPULATION_OPS = "DimensionManipulationOps"
-    OPSET_MERGE_OPS = "MergeOps"
     OPSET_CONV = "Conv"
     OPSET_DEPTHWISE_CONV = "DepthwiseConv2D"
     OPSET_CONV_TRANSPOSE = "ConvTraspose"
@@ -38,7 +33,10 @@ class OperatorSetNames(Enum):
     OPSET_GATHER = "Gather"
     OPSET_EXPAND = "Expend"
     OPSET_BATCH_NORM = "BatchNorm"
-    OPSET_ANY_RELU = "AnyReLU"
+    OPSET_RELU = "ReLU"
+    OPSET_RELU6 = "ReLU6"
+    OPSET_LEAKY_RELU = "LEAKYReLU"
+    OPSET_HARD_TANH = "HardTanh"
     OPSET_ADD = "Add"
     OPSET_SUB = "Sub"
     OPSET_MUL = "Mul"