chore: add data shape validation

Signed-off-by: Avik Basu <[email protected]>
numaproj · Jul 26, 2023 · 929d93b · 929d93b
1 parent 2da4ea0
commit 929d93b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/numalogic/models/threshold/_mahalanobis.py b/numalogic/models/threshold/_mahalanobis.py
@@ -21,6 +21,7 @@
 
 _INLIER: Final[int] = 0
 _OUTLIER: Final[int] = 1
+_INPUT_DIMS = Final[int] = 2
 
 
 class MahalanobisThreshold(BaseThresholdModel):
@@ -87,7 +88,7 @@ def _chebyshev_k(p: float) -> float:
     @staticmethod
     def _validate_input(x: npt.NDArray[float]) -> None:
         """Validate the input matrix shape."""
-        if x.ndim != 2:
+        if x.ndim != _INPUT_DIMS:
             raise InvalidDataShapeError(f"Input matrix should have 2 dims, given shape: {x.shape}.")
 
     def fit(self, x: npt.NDArray[float]) -> Self: