np<2 compat

Author: cwindolf

src/dartsort/cluster/gaussian_mixture.py

@@ -1879,6 +1879,8 @@ def get_log_likelihoods(
         # torch's index_select is painfully slow
         # weights = torch.index_select(likelihoods, 1, features.indices)
         # here we have weights as a csc_array
+        if torch.is_tensor(indices):
+            indices = indices.numpy(force=True)
         liks = likelihoods[:, indices]
         if unit_ids is not None:
             liks = liks[unit_ids]