Skip test_llama_tosa_MI if no input parameters are given

Author: mansnils

backends/arm/test/models/test_llama.py

@@ -61,7 +61,7 @@ def prepare_model(self):
             logging.warning(
                 "Skipping Llama test because of lack of input. To run use --llama_inputs <.pt> <.json>"
             )
-            return
+            return None, None, None
 
         assert os.path.isfile(checkpoint) and os.path.isfile(
             params_file
@@ -105,6 +105,9 @@ def prepare_model(self):
     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
+
         with torch.no_grad():
             (
                 ArmTester(