Convert mask len to int

src/smc/scheduler.py

@@ -142,7 +142,7 @@ class MeissonicScheduler(BaseScheduler):
             else:
                 raise ValueError(f"unknown masking schedule {self.masking_schedule}")
 
-            mask_len = (seq_len * mask_ratio).floor()
+            mask_len = (seq_len * mask_ratio).floor().long()
             # do not mask more than amount previously masked
             mask_len = torch.min(unknown_map.sum(dim=-1, keepdim=True) - 1, mask_len)
             # mask at least one