Skip to content

Commit 71719aa

Browse files
martinlsmMartin Lindström
authored andcommitted
Arm backend: Skip instead of returning in Llama test (#9906)
If the user does not provide input files with the --llama_inputs argument, the test is not run. Change the return statement into a pytest.skip so that the test output show a skip rather than a pass. Co-authored-by: Martin Lindström <[email protected]>
1 parent ed517f4 commit 71719aa

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

backends/arm/test/models/test_llama.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
import sys
1212
import unittest
1313

14+
import pytest
1415
import torch
1516

1617
from executorch.backends.arm.test import common, conftest
@@ -102,7 +103,7 @@ def test_llama_tosa_MI(self):
102103
llama_model, llama_inputs, llama_meta = self.prepare_model()
103104

104105
if llama_model is None and llama_inputs is None and llama_meta is None:
105-
return
106+
pytest.skip("Missing model and/or input files")
106107

107108
with torch.no_grad():
108109
(

0 commit comments

Comments
 (0)