diff --git a/utils/src/datasets.py b/utils/src/datasets.py
index 709fc430..4ab9a0bf 100644
--- a/utils/src/datasets.py
+++ b/utils/src/datasets.py
@@ -346,7 +346,7 @@ def collate(batch):
             out[key] = torch.nn.utils.rnn.pad_sequence(value, batch_first=True, padding_value=0)
             continue
         # raise Exception(f'Unknown batch key: {key}')
-        print(f"Passing {key}")
+        # print(f"Passing {key}")
 
     atom_mask = (out['fragment_mask'].bool() | out['linker_mask'].bool()).to(const.TORCH_INT)
     out['atom_mask'] = atom_mask[:, :, None]