From 19b1d363a5d0bc140b221ebba37b8eba8741a1ec Mon Sep 17 00:00:00 2001 From: Alex Yang <50511903+imabackstabber@users.noreply.github.com> Date: Mon, 9 May 2022 19:45:45 +0800 Subject: [PATCH] [Fix] Incorrectly append items to __all__ (#1944) --- mmcv/tensorrt/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mmcv/tensorrt/__init__.py b/mmcv/tensorrt/__init__.py index eb8cd4021..d86ddbf4b 100644 --- a/mmcv/tensorrt/__init__.py +++ b/mmcv/tensorrt/__init__.py @@ -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'])