Skip to content

Commit

Permalink
Exclude tests in CI
Browse files Browse the repository at this point in the history
  • Loading branch information
meatybobby committed Nov 25, 2024
1 parent b7bb214 commit 2a83e7d
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tests/export/test_trt_compile.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit 2a83e7d

Please sign in to comment.