Skip to content

Commit ce1cfcc

Browse files
authored
Add xfail to toy llama and deepseek tests (#1054)
For #1015.
1 parent 97d29ef commit ce1cfcc

File tree

2 files changed

+10
-0
lines changed

2 files changed

+10
-0
lines changed

sharktank/tests/models/deepseek/test_deepseek.py

+5
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,11 @@
1212
import torch
1313

1414

15+
@pytest.mark.xfail(
16+
reason="https://github.com/nod-ai/shark-ai/issues/1015",
17+
strict=False,
18+
raises=AssertionError,
19+
)
1520
def test_deepseek():
1621
theta, config = generate(12345)
1722
model = PagedDeepseekModelV1(theta=theta, config=config)

sharktank/tests/models/llama/test_llama.py

+5
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,11 @@
1212
import torch
1313

1414

15+
@pytest.mark.xfail(
16+
reason="https://github.com/nod-ai/shark-ai/issues/1015",
17+
strict=False,
18+
raises=AssertionError,
19+
)
1520
def test_llama():
1621
theta, config = generate(12345)
1722
model = PagedLlamaModelV1(theta=theta, config=config)

0 commit comments

Comments
 (0)