try fix windows

TimotheeMathieu · Nov 5, 2024 · ebcbd56 · ebcbd56
1 parent 9d34dce
commit ebcbd56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sklearn_extra/robust/_robust_weighted_estimator_helper.pyx b/sklearn_extra/robust/_robust_weighted_estimator_helper.pyx
@@ -70,7 +70,7 @@ cpdef np.ndarray[floating] _kmeans_loss(np.ndarray[floating, ndim=2, mode='c'] X
         np.ndarray[floating, ndim=2] centers = np.zeros([n_classes,
                                                          n_features],
                                                          dtype = dtype)
-        np.ndarray[int] num_in_cluster = np.zeros(n_classes, dtype = int)
+        np.ndarray[np.int] num_in_cluster = np.zeros(n_classes, dtype = np.int)
         np.ndarray[floating] inertias = np.zeros(n_samples, dtype = dtype)
     for i in range(n_samples):
         for j in range(n_features):