Update datasets.py

globus-labs · Sep 3, 2023 · b51ce27 · b51ce27
1 parent 20ea830
commit b51ce27
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]