clean up todo

choderalab · Jan 31, 2024 · 48721fb · 48721fb
1 parent e8fe374
commit 48721fb
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/mtenn/conversion_utils/visnet.py b/mtenn/conversion_utils/visnet.py
@@ -85,7 +85,6 @@ def forward(self, data):
             z = data["z"]
 
             # all atom in one pass from the same molecule
-            # TODO: set separate batch for ligand and protein
             batch = torch.zeros(z.shape[0], device=z.device)
             x, v = self.visnet.representation_model(z, pos, batch)
             x = self.visnet.output_model.pre_reduce(x, v)