fix path at test_onnx_trt_export

garciadias · garciadias · commit 75afad0dc5f4 · 2025-02-21T09:51:16.000Z
diff --git a/tests/bundle/test_bundle_trt_export.py b/tests/bundle/test_bundle_trt_export.py
@@ -108,7 +108,7 @@ def test_trt_export(self, convert_precision, input_shape, dynamic_batch):
         has_onnx and has_torchtrt and has_tensorrt, "Onnx and TensorRT are required for onnx-trt conversion!"
     )
     def test_onnx_trt_export(self, convert_precision, input_shape, dynamic_batch):
-        tests_dir = Path(__file__).resolve().parent
+        tests_dir = Path(__file__).resolve().parents[1]
         meta_file = os.path.join(tests_dir, "testing_data", "metadata.json")
         config_file = os.path.join(tests_dir, "testing_data", "inference.json")
         with tempfile.TemporaryDirectory() as tempdir:

Original file line number	Diff line number	Diff line change
`@@ -108,7 +108,7 @@ def test_trt_export(self, convert_precision, input_shape, dynamic_batch):`
`108`	`108`	`has_onnx and has_torchtrt and has_tensorrt, "Onnx and TensorRT are required for onnx-trt conversion!"`
`109`	`109`	`)`
`110`	`110`	`def test_onnx_trt_export(self, convert_precision, input_shape, dynamic_batch):`
`111`		`- tests_dir = Path(__file__).resolve().parent`
	`111`	`+ tests_dir = Path(__file__).resolve().parents[1]`
`112`	`112`	`meta_file = os.path.join(tests_dir, "testing_data", "metadata.json")`
`113`	`113`	`config_file = os.path.join(tests_dir, "testing_data", "inference.json")`
`114`	`114`	`with tempfile.TemporaryDirectory() as tempdir:`