update

microsoft · Nov 10, 2023 · 0bd5b8c · 0bd5b8c
1 parent f9206f3
commit 0bd5b8c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/onnxruntime/test/python/onnxruntime_test_engine_wrapper.py b/onnxruntime/test/python/onnxruntime_test_engine_wrapper.py
@@ -18,10 +18,9 @@ class TestInferenceSessionWithCtxNode(unittest.TestCase):
 
     # This test is only for TRT EP to test EPContext node with TRT engine
     @unittest.skipIf(
-        "TensorrtExecutionProvider" not in ort.get_available_providers(),
+        "TensorrtExecutionProvider" not in onnxrt.get_available_providers(),
         reason="Test TRT EP only",
     )
-
     def create_ctx_node(self, ctx_embed_mode=0, cache_path=""):
         if ctx_embed_mode:
             # Get engine buffer from engine cache