Remove cache option from njit decorators in ImputeMultiVariate class

CyrilJl · Dec 25, 2024 · d659149 · d659149
1 parent c0354fe
commit d659149
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/timefiller/_multivariate_imputer.py b/timefiller/_multivariate_imputer.py
@@ -120,7 +120,7 @@ def _prepare_data(cls, mask_nan, col_to_impute, subset_rows):
         return index_predict, columns
 
     @staticmethod
-    @njit(parallel=True, fastmath=True, boundscheck=False, cache=True)
+    @njit(parallel=True, fastmath=True, boundscheck=False)
     def _split(X, index_predict, selected_rows, selected_cols, col_to_impute):
         """
         Splits the data into training and prediction sets.
@@ -151,7 +151,7 @@ def _split(X, index_predict, selected_rows, selected_cols, col_to_impute):
         return X_train, y_train, X_pred
 
     @staticmethod
-    @njit(parallel=True, fastmath=True, boundscheck=False, cache=True)
+    @njit(parallel=True, fastmath=True, boundscheck=False)
     def _subset(X, rows, columns):
         """
         Extracts a subset from the array X.