Update the doc string (openmm#44)

raimis · Dec 20, 2021 · f3a93d3 · f3a93d3
1 parent b0383c0
commit f3a93d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pytorch/OptimizedTorchANI.py b/src/pytorch/OptimizedTorchANI.py
@@ -32,7 +32,7 @@
 
 class OptimizedTorchANI(torch.nn.Module):
 
-    from torchani.models import BuiltinModel
+    from torchani.models import BuiltinModel # https://github.com/openmm/NNPOps/issues/44
 
     def __init__(self, model: BuiltinModel, atomicNumbers: Tensor) -> None: