Skip to content

Commit f799b57

Browse files
authored
Update test_prompter.py
1 parent b6ca41c commit f799b57

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/contrib/test_prompter.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def test_gradcheck(self, device):
8787
def test_module(self):
8888
...
8989

90-
@pytest.mark.skipif(torch_version() in {"2.0.0", "2.0.1"}, reason="Not working on 2.0")
90+
@pytest.mark.skipif(torch_version() in {"2.0.0", "2.0.1", "2.1.2"}, reason="Not working on 2.0.x and 2.1.x")
9191
def test_dynamo(self, device, torch_optimizer):
9292
dtype = torch.float32
9393
batch_size = 1

0 commit comments

Comments
 (0)