clean up print statement (#104)

talmolab · Dec 19, 2024 · 08256ae · 08256ae
1 parent d79f2f7
commit 08256ae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dreem/inference/post_processing.py b/dreem/inference/post_processing.py
@@ -159,7 +159,7 @@ def filter_max_center_dist(
 
         valid = dist.squeeze() < max_center_dist  # n_k x n_nonk
         valid_mult = valid.float().unsqueeze(-1) if valid.ndim == 1 else valid.float()
-        print(dist.shape, valid_mult.shape, id_inds.shape)
+        # print(dist.shape, valid_mult.shape, id_inds.shape)
         valid_assn = (
             torch.mm(valid_mult, id_inds.to(valid.device)).clamp_(max=1.0).long().bool()
         )  # n_k x M