We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c70141c commit ea8d0fdCopy full SHA for ea8d0fd
tests/test_inference.py
@@ -38,11 +38,15 @@
38
models_to_test = list(models_to_test)
39
models_to_test.sort()
40
41
+# Keep (model_name, False) tuples so test IDs remain stable for CI caching
42
+models_to_test_fully = [(m, False) for m in models_to_test]
43
+
44
45
@pytest.mark.regression_test
-@pytest.mark.parametrize("model_name", models_to_test)
46
+@pytest.mark.parametrize("model_name,jit", models_to_test_fully)
47
def test_inference_with_data(
48
model_name,
49
+ jit,
50
pretrained = None,
51
pretrained_hf = False,
52
precision = 'fp32',
0 commit comments