Fix importer in test_yolo_trt_module

zhiqwang · Dec 24, 2021 · ec6a4ec · ec6a4ec
1 parent 0dcb60a
commit ec6a4ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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