Accidently dropped pytestmark from other test

f19a89a7 · Logan Adams · ae7166af · f19a89a7
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

tests/unit/inference/test_inference.py tests/unit/inference/test_inference.py +1 -0

未找到文件。
--- a/tests/unit/inference/test_inference.py
+++ b/tests/unit/inference/test_inference.py
@@ -425,6 +425,7 @@ class TestLowCpuMemUsage(DistributedTest):
        assert assert_fn(bs_output, ds_output)


+@pytest.mark.seq_inference
 @pytest.mark.parametrize("model_w_task", [("tiiuae/falcon-7b", "text-generation")], ids=["falcon"])
 class TestAutoTP(DistributedTest):
    world_size = 1