diff --git a/backends/arm/test/models/test_llama.py b/backends/arm/test/models/test_llama.py index bd18ff1856f..89196674c48 100644 --- a/backends/arm/test/models/test_llama.py +++ b/backends/arm/test/models/test_llama.py @@ -11,6 +11,7 @@ import sys import unittest +import pytest import torch from executorch.backends.arm.test import common, conftest @@ -102,7 +103,7 @@ def test_llama_tosa_MI(self): llama_model, llama_inputs, llama_meta = self.prepare_model() if llama_model is None and llama_inputs is None and llama_meta is None: - return + pytest.skip("Missing model and/or input files") with torch.no_grad(): (