From 3b0c04d2e8376ea69104de297062e8b91233dece Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Martin=20Lindstr=C3=B6m?= Date: Wed, 26 Mar 2025 13:45:00 +0100 Subject: [PATCH] Arm backend: Skip instead of returning in Llama test 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. Change-Id: I5a1836cb9a16dc5b3fd43e25f1586d901cc8e8f9 --- backends/arm/test/models/test_llama.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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(): (