Revised setcol() for csr

myui · Dec 10, 2024 · f38cb47 · f38cb47
1 parent 73ccd1e
commit f38cb47
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/rtrec/models/internal/slim_elastic.py b/rtrec/models/internal/slim_elastic.py
@@ -96,7 +96,9 @@ def set_col(self, j: int, values: ArrayLike) -> None:
         values : ArrayLike
             The new values for the column.
         """
-        self.csc_matrix[:, j] = values
+        start, end = self.csc_matrix.indptr[j], self.csc_matrix.indptr[j+1]
+        assert len(values) == end - start, f"Values must have the same length as the column: {len(values)} != {end - start}"
+        self.csc_matrix.data[start:end] = values
 
 class FeatureSelectionWrapper: