Add missing parse_args from previous bad merge (#241)

* Add missing parse_args from previous bad merge * Test torchmd-train in CI
torchmd · Nov 20, 2023 · 0275e54 · 0275e54
1 parent be6a214
commit 0275e54
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
@@ -44,3 +44,7 @@ jobs:
 
       - name: Run tests
         run: pytest -v -s
+
+      - name: Test torchmd-train utility
+        run: torchmd-train --help
+
diff --git a/torchmdnet/scripts/train.py b/torchmdnet/scripts/train.py
@@ -108,8 +108,8 @@ def get_argparse():
 
 def get_args():
 
-    args = get_argparse()
-
+    parser = get_argparse()
+    args = parser.parse_args()
     if args.redirect:
         sys.stdout = open(os.path.join(args.log_dir, "log"), "w")
         sys.stderr = sys.stdout