FIX errors

interTwin-eu · Apr 30, 2024 · ce8df42 · ce8df42
1 parent 1ef12e7
commit ce8df42
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/tests/use-cases/test_cyclones.py b/tests/use-cases/test_cyclones.py
@@ -27,6 +27,7 @@ def test_cyclones_train_tf(tf_env, tmp_test_dir, install_requirements):
     # TODO: create a small sample dataset for tests only
     install_requirements(CYCLONES_PATH, tf_env)
     pipe = os.path.join(os.path.abspath(CYCLONES_PATH), 'pipeline.yaml')
-    cmd = (f"{tf_env}/bin/python train.py "
+    train = os.path.join(os.path.abspath(CYCLONES_PATH), 'train.py')
+    cmd = (f"{tf_env}/bin/python {train} "
            f"-p {pipe}")
     subprocess.run(cmd.split(), check=True, cwd=tmp_test_dir)
diff --git a/use-cases/3dgan/create_inference_sample.py b/use-cases/3dgan/create_inference_sample.py
@@ -20,4 +20,4 @@ def create_checkpoint(
     parser.add_argument("--root", type=str, default='.')
     parser.add_argument("--ckpt-name", type=str, default="3dgan-inference.pth")
     args = parser.parse_args()
-    create_checkpoint(vars(args))
+    create_checkpoint(**vars(args))