From 9bbb2deaf5cdd26a7e6a5612eb31f3a6d6762bf6 Mon Sep 17 00:00:00 2001 From: Timothee Mathieu Date: Tue, 5 Nov 2024 13:35:09 +0100 Subject: [PATCH] try fix windows --- sklearn_extra/robust/_robust_weighted_estimator_helper.pyx | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/sklearn_extra/robust/_robust_weighted_estimator_helper.pyx b/sklearn_extra/robust/_robust_weighted_estimator_helper.pyx index d05945cc..b1b781aa 100644 --- a/sklearn_extra/robust/_robust_weighted_estimator_helper.pyx +++ b/sklearn_extra/robust/_robust_weighted_estimator_helper.pyx @@ -14,6 +14,7 @@ import sys from time import time from libc.math cimport exp, log, sqrt, pow, fabs +from libc.stdint cimport int64_t cimport numpy as np from numpy.math cimport INFINITY @@ -66,7 +67,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[long] num_in_cluster = np.zeros(n_classes, dtype = int) + np.ndarray[int64_t] num_in_cluster = np.zeros(n_classes, dtype = int) np.ndarray[floating] inertias = np.zeros(n_samples, dtype = dtype) for i in range(n_samples): for j in range(n_features):