From fc1479b16320417ae232532fd0c40a90e5dda589 Mon Sep 17 00:00:00 2001 From: Luca Freckmann Date: Fri, 14 Feb 2025 10:59:52 +0100 Subject: [PATCH 1/2] scaler does not rescale if scale is 1 or 1.0 --- synapse_net/inference/util.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/synapse_net/inference/util.py b/synapse_net/inference/util.py index d6a6abc..12883fa 100644 --- a/synapse_net/inference/util.py +++ b/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 From 0548faf61e672c1c6d62bc8fe73d3fc7f57a5f70 Mon Sep 17 00:00:00 2001 From: Luca Date: Fri, 14 Feb 2025 11:25:59 +0100 Subject: [PATCH 2/2] linter updates --- synapse_net/inference/util.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/synapse_net/inference/util.py b/synapse_net/inference/util.py index 12883fa..4545d6f 100644 --- a/synapse_net/inference/util.py +++ b/synapse_net/inference/util.py @@ -33,9 +33,9 @@ class _Scaler: def __init__(self, scale, verbose): - if isinstance(scale, (int, float)): # single number + if isinstance(scale, (int, float)): # single number is_one = scale in (1, 1.0) - else: # list + else: # list is_one = all(s in (1, 1.0) for s in scale) if is_one: self.scale = None