From ec6a4ecffdd801e8b4576f4c931295431867aa5d Mon Sep 17 00:00:00 2001
From: Zhiqiang Wang <zhiqwang@foxmail.com>
Date: Sat, 25 Dec 2021 02:23:44 +0800
Subject: [PATCH] Fix importer in test_yolo_trt_module

---
 test/test_runtime.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/test_runtime.py b/test/test_runtime.py
index b1e4692a..f721d0ab 100644
--- a/test/test_runtime.py
+++ b/test/test_runtime.py
@@ -4,7 +4,7 @@
 import pytest
 import torch
 from torch import Tensor
-from yolort.runtime import YOLOTRTModule
+from yolort.runtime.yolo_tensorrt_model import YOLOTRTModule
 from yolort.v5 import attempt_download