Skip to content

GitLab

Unverified Commit 19b1d363 authored by Alex Yang's avatar Alex Yang Committed by GitHub
Browse files

[Fix] Incorrectly append items to __all__ (#1944)

parent e70a318a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
mmcv/tensorrt/__init__.py
View file @ 19b1d363
...@@ -22,9 +22,9 @@ if is_tensorrt_available(): ...@@ -22,9 +22,9 @@ if is_tensorrt_available():
# load tensorrt plugin lib # load tensorrt plugin lib
load_tensorrt_plugin() load_tensorrt_plugin()
__all__.append([ __all__.extend([
'onnx2trt', 'save_trt_engine', 'load_trt_engine', 'TRTWraper', 'onnx2trt', 'save_trt_engine', 'load_trt_engine', 'TRTWraper',
'TRTWrapper' 'TRTWrapper'
]) ])
__all__.append(['is_tensorrt_plugin_loaded', 'preprocess_onnx']) __all__.extend(['is_tensorrt_plugin_loaded', 'preprocess_onnx'])
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment