Skip to content

Commit da29d2f

Browse files
committed
Corrects test paths target
1 parent c64825f commit da29d2f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/bundle/test_bundle_trt_export.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def tearDown(self):
7070
@parameterized.expand([TEST_CASE_1, TEST_CASE_2, TEST_CASE_3, TEST_CASE_4])
7171
@unittest.skipUnless(has_torchtrt and has_tensorrt, "Torch-TensorRT is required for conversion!")
7272
def test_trt_export(self, convert_precision, input_shape, dynamic_batch):
73-
tests_dir = Path(__file__).resolve().parent
73+
tests_dir = Path(__file__).resolve().parents[1]
7474
meta_file = os.path.join(tests_dir, "testing_data", "metadata.json")
7575
config_file = os.path.join(tests_dir, "testing_data", "inference.json")
7676
with tempfile.TemporaryDirectory() as tempdir:

0 commit comments

Comments
 (0)