mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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

pull/1945/head
Alex Yang 2022-05-09 19:45:45 +08:00 committed by GitHub
parent e70a318ae9
commit 19b1d363a5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mmcv/tensorrt/__init__.py
View File

@ -22,9 +22,9 @@ if is_tensorrt_available():
# load tensorrt plugin lib
load_tensorrt_plugin()
__all__.append([
__all__.extend([
'onnx2trt', 'save_trt_engine', 'load_trt_engine', 'TRTWraper',
'TRTWrapper'
])
__all__.append(['is_tensorrt_plugin_loaded', 'preprocess_onnx'])
__all__.extend(['is_tensorrt_plugin_loaded', 'preprocess_onnx'])