Skip to content

Commit

Permalink
lintrunner -a
Browse files Browse the repository at this point in the history
  • Loading branch information
chilo-ms committed Jan 24, 2024
1 parent bb79ca9 commit 0e7334e
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion onnxruntime/python/onnxruntime_inference_collection.py
Original file line number Diff line number Diff line change
Expand Up @@ -514,7 +514,11 @@ def _register_ep_custom_ops(self, session_options, providers, provider_options,
for i in range(len(providers)):
if providers[i] in available_providers and providers[i] == "TensorrtExecutionProvider":
C.register_tensorrt_plugins_as_custom_ops(session_options, provider_options[i])
elif isinstance(providers[i], tuple) and providers[i][0] in available_providers and providers[i][0] == "TensorrtExecutionProvider":
elif (
isinstance(providers[i], tuple)
and providers[i][0] in available_providers
and providers[i][0] == "TensorrtExecutionProvider"
):
C.register_tensorrt_plugins_as_custom_ops(session_options, providers[i][1])


Expand Down

0 comments on commit 0e7334e

Please sign in to comment.