Skip to content

Commit 9bbb2de

Browse files
try fix windows
1 parent 0aa5ef7 commit 9bbb2de

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

sklearn_extra/robust/_robust_weighted_estimator_helper.pyx

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ import sys
1414
from time import time
1515

1616
from libc.math cimport exp, log, sqrt, pow, fabs
17+
from libc.stdint cimport int64_t
1718
cimport numpy as np
1819
from numpy.math cimport INFINITY
1920

@@ -66,7 +67,7 @@ cpdef np.ndarray[floating] _kmeans_loss(np.ndarray[floating, ndim=2, mode='c'] X
6667
np.ndarray[floating, ndim=2] centers = np.zeros([n_classes,
6768
n_features],
6869
dtype = dtype)
69-
np.ndarray[long] num_in_cluster = np.zeros(n_classes, dtype = int)
70+
np.ndarray[int64_t] num_in_cluster = np.zeros(n_classes, dtype = int)
7071
np.ndarray[floating] inertias = np.zeros(n_samples, dtype = dtype)
7172
for i in range(n_samples):
7273
for j in range(n_features):

0 commit comments

Comments
 (0)