fixed paths to scripts module

torchmd · Feb 12, 2025 · ffd4c89 · ffd4c89
1 parent da2ab6e
commit ffd4c89
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/source/models.rst b/docs/source/models.rst
@@ -342,7 +342,7 @@ To implement a new architecture, you need to follow these steps:
                 **shared_args,
             )
 
-4. Add any new parameters required to initialize your module to scripts.train.get_args:
+4. Add any new parameters required to initialize your module to torchmdnet.scripts.train.get_args:
 
     .. code-block:: python
 

diff --git a/docs/source/torchmd-train.rst b/docs/source/torchmd-train.rst
@@ -89,7 +89,7 @@ Command line interface
 ~~~~~~~~~~~~~~~~~~~~~~
 
 
-.. autoprogram:: scripts.train:get_argparse()
+.. autoprogram:: torchmdnet.scripts.train:get_argparse()
    :prog: torchmd-train