fix typo in docstring

choderalab · Feb 7, 2024 · d14f65d · d14f65d
1 parent 5f161a3
commit d14f65d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mtenn/conversion_utils/visnet.py b/mtenn/conversion_utils/visnet.py
@@ -170,13 +170,13 @@ def get_model(
             comb_readout=None,
         ):
             """
-            Exposed function to build a Model object from a SchNet object. If none
+            Exposed function to build a Model object from a VisNet object. If none
             is provided, a default model is initialized.
 
             Parameters
             ----------
-            model: SchNet, optional
-                SchNet model to use to build the Model object. If left as none, a
+            model: VisNet, optional
+                VisNet model to use to build the Model object. If left as none, a
                 default model will be initialized and used
             grouped: bool, default=False
                 Whether this model should accept groups of inputs or one input at a