Fix import

borisfom · Nov 26, 2024 · 5111223 · 5111223
1 parent b7e1958
commit 5111223
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/export/test_trt_compile.py b/tests/export/test_trt_compile.py
@@ -15,6 +15,8 @@
 import unittest
 from typing import List
 
+import torch
+
 TEST_CASE_1 = ["fp32"]
 TEST_CASE_2 = ["fp16"]