Fix same min/max values

synnax-ai · Aug 30, 2024 · 55cfd84 · 55cfd84
1 parent 49d4b96
commit 55cfd84
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "synnax-shared"
-version = "1.7.2"
+version = "1.7.3"
 description = "Synnax shared Python pacakges"
 readme = "README.md"
 requires-python = ">=3.11"

diff --git a/src/synnax_shared/data_processing/scaler/min_max_scaler.py b/src/synnax_shared/data_processing/scaler/min_max_scaler.py
@@ -15,9 +15,15 @@ def __init__(self, min: float | None = None, max: float | None = None):
         self.max = max
 
     def fit_transform(self, ndarray: NDArray) -> NDArray:
+        if self.min is not None or self.max is not None:
+            raise ValueError("MinMaxScaler is already fitted")
         float_ndarray = ndarray.astype(float)
-        self.min = numpy.nanmin(float_ndarray)
-        self.max = numpy.nanmax(float_ndarray)
+        min = numpy.nanmin(float_ndarray)
+        max = numpy.nanmax(float_ndarray)
+        if min == max:
+            min = max - 1
+        self.min = min
+        self.max = max
         return self.transform(float_ndarray)
 
     def transform(self, ndarray: NDArray) -> NDArray: