Fix: Skip emitted medoids

RasmussenLab · Sep 14, 2023 · e0a86fd · e0a86fd
1 parent b621f54
commit e0a86fd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/vamb/cluster.py b/vamb/cluster.py
@@ -384,6 +384,7 @@ def get_next_seed(self) -> int:
             if (
                 new_index >= len(self.indices)
                 or self.indices[new_index].item() != order
+                or (self.cuda and not self.kept_mask[new_index].item())
             ):
                 self.order[i] = -1
                 continue