Skip to content

Commit ea8d0fd

Browse files
committed
Keep jit for inference test data cache
1 parent c70141c commit ea8d0fd

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tests/test_inference.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,11 +38,15 @@
3838
models_to_test = list(models_to_test)
3939
models_to_test.sort()
4040

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+
4144

4245
@pytest.mark.regression_test
43-
@pytest.mark.parametrize("model_name", models_to_test)
46+
@pytest.mark.parametrize("model_name,jit", models_to_test_fully)
4447
def test_inference_with_data(
4548
model_name,
49+
jit,
4650
pretrained = None,
4751
pretrained_hf = False,
4852
precision = 'fp32',

0 commit comments

Comments
 (0)