Merge pull request #10 from ashwinvaidya17/fix/update_tests_2

fix path
samet-akcay · Feb 29, 2024 · e04b85d · e04b85d
2 parents 137d726 + 5029f04
commit e04b85d
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 11 deletions.
diff --git a/tests/integration/tools/test_gradio_entrypoint.py b/tests/integration/tools/test_gradio_entrypoint.py
@@ -47,14 +47,14 @@ def test_torch_inference(
         # export torch model
         export_to_torch(
             model=model,
-            export_root=_ckpt_path.parent.parent,
+            export_root=_ckpt_path.parent.parent.parent,
             task=TaskType.SEGMENTATION,
         )
 
         arguments = parser().parse_args(
             [
                 "--weights",
-                str(_ckpt_path.parent) + "/torch/model.pt",
+                str(_ckpt_path.parent.parent) + "/torch/model.pt",
             ],
         )
         assert isinstance(inferencer(arguments.weights, arguments.metadata), TorchInferencer)
@@ -71,7 +71,7 @@ def test_openvino_inference(
 
         # export OpenVINO model
         export_to_openvino(
-            export_root=_ckpt_path.parent.parent,
+            export_root=_ckpt_path.parent.parent.parent,
             model=model,
             ov_args={},
             task=TaskType.SEGMENTATION,
@@ -80,9 +80,9 @@ def test_openvino_inference(
         arguments = parser().parse_args(
             [
                 "--weights",
-                str(_ckpt_path.parent) + "/openvino/model.bin",
+                str(_ckpt_path.parent.parent) + "/openvino/model.bin",
                 "--metadata",
-                str(_ckpt_path.parent) + "/openvino/metadata.json",
+                str(_ckpt_path.parent.parent) + "/openvino/metadata.json",
             ],
         )
         assert isinstance(inferencer(arguments.weights, arguments.metadata), OpenVINOInferencer)

diff --git a/tests/integration/tools/test_openvino_entrypoint.py b/tests/integration/tools/test_openvino_entrypoint.py
@@ -44,7 +44,7 @@ def test_openvino_inference(
 
         # export OpenVINO model
         export_to_openvino(
-            export_root=_ckpt_path.parent.parent,
+            export_root=_ckpt_path.parent.parent.parent,
             model=model,
             ov_args={},
             task=TaskType.SEGMENTATION,
@@ -53,13 +53,13 @@ def test_openvino_inference(
         arguments = get_parser().parse_args(
             [
                 "--weights",
-                str(_ckpt_path.parent) + "/openvino/model.bin",
+                str(_ckpt_path.parent.parent) + "/openvino/model.bin",
                 "--metadata",
-                str(_ckpt_path.parent) + "/openvino/metadata.json",
+                str(_ckpt_path.parent.parent) + "/openvino/metadata.json",
                 "--input",
                 get_dummy_inference_image,
                 "--output",
-                str(_ckpt_path.parent) + "/output.png",
+                str(_ckpt_path.parent.parent) + "/output.png",
             ],
         )
         infer(arguments)
diff --git a/tests/integration/tools/test_torch_entrypoint.py b/tests/integration/tools/test_torch_entrypoint.py
@@ -44,13 +44,13 @@ def test_torch_inference(
         model = Padim.load_from_checkpoint(_ckpt_path)
         export_to_torch(
             model=model,
-            export_root=_ckpt_path.parent.parent,
+            export_root=_ckpt_path.parent.parent.parent,
             task=TaskType.SEGMENTATION,
         )
         arguments = get_parser().parse_args(
             [
                 "--weights",
-                str(_ckpt_path.parent) + "/torch/model.pt",
+                str(_ckpt_path.parent.parent) + "/torch/model.pt",
                 "--input",
                 get_dummy_inference_image,
                 "--output",