From 15b4abf5c7fbbeab2b14abe46e288a8d4666b540 Mon Sep 17 00:00:00 2001
From: Malte Londschien <61679398+mlondschien@users.noreply.github.com>
Date: Wed, 18 Sep 2024 10:56:31 +0200
Subject: [PATCH] Update src/tabmat/categorical_matrix.py

Co-authored-by: Luca Bittarello <15511539+lbittarello@users.noreply.github.com>
---
 src/tabmat/categorical_matrix.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/tabmat/categorical_matrix.py b/src/tabmat/categorical_matrix.py
index a7c25e19..a0d0c2fa 100644
--- a/src/tabmat/categorical_matrix.py
+++ b/src/tabmat/categorical_matrix.py
@@ -674,8 +674,7 @@ def _get_col_stds(self, weights: np.ndarray, col_means: np.ndarray) -> np.ndarra
         mean = self.transpose_matvec(weights)
         vars = mean - col_means**2
         # If using float32, we can get negative values due to precision errors
-        vars[vars < 0] = 0
-        return np.sqrt(vars)
+        return np.sqrt(np.maximum(vars, 0))
 
     def __getitem__(self, item):
         row, col = _check_indexer(item)