From 1d00be97b3b2176ffbbf714227b530fb2bdd9a88 Mon Sep 17 00:00:00 2001 From: knc6 Date: Wed, 27 Nov 2024 16:35:55 -0500 Subject: [PATCH] Fix test --- atomgpt/tests/test_forward.py | 6 +++--- atomgpt/tests/test_inverse.py | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/atomgpt/tests/test_forward.py b/atomgpt/tests/test_forward.py index 1560e11..658448f 100644 --- 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 index 55abaf9..58ed225 100644 --- 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