Fix compressor options (#279)

Nota-NetsPresso · Jul 9, 2024 · aa5cded · aa5cded
1 parent 1150a20
commit aa5cded
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/netspresso/VERSION b/netspresso/VERSION
@@ -1 +1 @@
-1.7.1.b3
+1.7.1
diff --git a/netspresso/clients/compressor/v2/schemas/compression/base.py b/netspresso/clients/compressor/v2/schemas/compression/base.py
@@ -16,6 +16,7 @@ def __post_init__(self):
             )
 
 
+@dataclass
 class Options(OptionsBase):
     policy: Policy = Policy.AVERAGE
     layer_norm: LayerNorm = LayerNorm.STANDARD_SCORE
@@ -25,6 +26,7 @@ class Options(OptionsBase):
     reverse: bool = False
 
 
+@dataclass
 class RecommendationOptions(Options):
     min_num_of_value: int = 8