scaler does not rescale if scale is 1 or 1.0

Author: lufre1

synapse_net/inference/util.py

@@ -33,7 +33,14 @@
 
 class _Scaler:
     def __init__(self, scale, verbose):
-        self.scale = scale
+        if isinstance(scale, (int, float)): # single number
+            is_one = scale in (1, 1.0)
+        else: # list
+            is_one = all(s in (1, 1.0) for s in scale)
+        if is_one:
+            self.scale = None
+        else:
+            self.scale = scale
         self.verbose = verbose
         self._original_shape = None