From 7215fb4a24ebb6f163b4922fe83c582b8b899d07 Mon Sep 17 00:00:00 2001
From: Zhiqiang Wang <zhiqwang@foxmail.com>
Date: Sun, 26 Dec 2021 18:45:19 +0800
Subject: [PATCH] Minor fixes

---
 yolort/runtime/trt_helper.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/yolort/runtime/trt_helper.py b/yolort/runtime/trt_helper.py
index eb3dabad..16bf9c06 100644
--- a/yolort/runtime/trt_helper.py
+++ b/yolort/runtime/trt_helper.py
@@ -31,6 +31,9 @@
 logger = logging.getLogger("TRTHelper")
 
 
+__all__ = ["YOLOTRTModule", "EngineBuilder"]
+
+
 class YOLOTRTModule(nn.Module):
     """
     TensorRT deployment friendly wrapper for YOLO.