fix filepath in tests

issue from folder refactoring
ivy-llc · Jun 6, 2023 · 45a5462 · 45a5462
1 parent a06d970
commit 45a5462
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ivy_models_tests/resnet/test_resnet.py b/ivy_models_tests/resnet/test_resnet.py
@@ -32,7 +32,7 @@ def test_resnet_18_img_classification(device, f, fw, batch_shape, load_weights):
         this_dir = os.path.dirname(os.path.realpath(__file__))
         weight_fpath = os.path.join(
             this_dir,
-            "../ivy_models/resnet/pretrained_weights/resnet_18.pickled",
+            "../../ivy_models/resnet/pretrained_weights/resnet_18.pickled",
         )
 
         # Check if weight file exists

diff --git a/ivy_models_tests/transformers/test_transformers.py b/ivy_models_tests/transformers/test_transformers.py
@@ -80,7 +80,7 @@ def test_perceiver_io_img_classification(
         this_dir = os.path.dirname(os.path.realpath(__file__))
         weight_fpath = os.path.join(
             this_dir,
-            "../ivy_models/transformers/pretrained_weights/perceiver_io.pickled",
+            "../../ivy_models/transformers/pretrained_weights/perceiver_io.pickled",
         )
         assert os.path.isfile(weight_fpath)
         # noinspection PyBroadException