Skip to content

Commit 309939b

Browse files
committed
re-add np.int64 support for classic PAM
1 parent 2e3de9f commit 309939b

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

kmedoids/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -362,6 +362,8 @@ def pam(diss, medoids, max_iter=100, init="build", random_state=None):
362362
return KMedoidsResult(*_pam_swap_f64(diss, medoids.astype(np.uint64), max_iter))
363363
elif dtype == np.int32:
364364
return KMedoidsResult(*_pam_swap_i32(diss, medoids.astype(np.uint64), max_iter))
365+
elif dtype == np.int64:
366+
return KMedoidsResult(*_pam_swap_i64(diss, medoids.astype(np.uint64), max_iter))
365367
raise ValueError("Input data not supported. Use a numpy array of floats.")
366368

367369
def pammedsil(diss, medoids, max_iter=100, init="build", random_state=None):

0 commit comments

Comments
 (0)