From 511122399652d0a5eff775ddd9d848be5a5640ff Mon Sep 17 00:00:00 2001
From: Bobby Chen <bobchen@nvidia.com>
Date: Tue, 26 Nov 2024 10:51:26 -0800
Subject: [PATCH] Fix import

---
 tests/export/test_trt_compile.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/export/test_trt_compile.py b/tests/export/test_trt_compile.py
index 990bdb8ec532..09a77004678f 100644
--- 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"]