diff --git a/tests/export/test_trt_compile.py b/tests/export/test_trt_compile.py index 95be5f88bf01..24dc0bcd2ef0 100644 --- a/tests/export/test_trt_compile.py +++ b/tests/export/test_trt_compile.py @@ -42,6 +42,7 @@ def forward(self, x: List[torch.Tensor], y: torch.Tensor, z: torch.Tensor, bs: f return x1, [y1, z1], y1 + z1 +@unittest.skip @unittest.skipUnless(trt_imported, "tensorrt is required") @unittest.skipUnless(polygraphy_imported, "polygraphy is required") class TestTRTCompile(unittest.TestCase):