Fix test

usnistgov · Nov 27, 2024 · 1d00be9 · 1d00be9
1 parent aa1bb73
commit 1d00be9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/atomgpt/tests/test_forward.py b/atomgpt/tests/test_forward.py
@@ -1,11 +1,11 @@
 import os
-from atomgpt.forward_models.forward_models import run_atomgpt
+from atomgpt.forward_models.forward_models import main
 
 
 def test_forward():
-    run_atomgpt("atomgpt/examples/forward_model/config.json")
+    main("atomgpt/examples/forward_model/config.json")
     # cmd='python atomgpt/forward_models/forward_models.py --config_name atomgpt/examples/forward_model/config.json'
     # os.system(cmd)
 
 
-# test_forward()
+#test_forward()
diff --git a/atomgpt/tests/test_inverse.py b/atomgpt/tests/test_inverse.py
@@ -1,4 +1,4 @@
-from atomgpt.inverse_models.inverse_models import run_atomgpt_inverse
+from atomgpt.inverse_models.inverse_models import main
 import os
 import atomgpt
 
@@ -10,7 +10,7 @@
 
 def run_inverse():
     try:
-        run_atomgpt_inverse(config)
+        main(config)
     except Exception as exp:
         print("exp", exp)
         pass