diff --git a/tests/export/test_trt_compile.py b/tests/export/test_trt_compile.py index 520b3f0a56a1..990bdb8ec532 100644 --- a/tests/export/test_trt_compile.py +++ b/tests/export/test_trt_compile.py @@ -15,11 +15,6 @@ import unittest from typing import List -import torch - -from nemo.export import trt_compile -from nemo.utils.import_utils import safe_import - TEST_CASE_1 = ["fp32"] TEST_CASE_2 = ["fp16"]