From 1a81863b62c2a8a465729dd8ce49c67548751a51 Mon Sep 17 00:00:00 2001
From: kkeroo <61207502+kkeroo@users.noreply.github.com>
Date: Thu, 9 Jan 2025 16:38:29 +0100
Subject: [PATCH] Improved logging.

---
 tests/end_to_end/main.py     | 4 ++--
 tests/end_to_end/test_e2e.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/end_to_end/main.py b/tests/end_to_end/main.py
index 054c823f..d13b1f70 100644
--- a/tests/end_to_end/main.py
+++ b/tests/end_to_end/main.py
@@ -69,7 +69,7 @@ def main():
         f"--nn_archive_path={nn_archive_path}",
         f"--platform={rvc_platform}",
         "-v",
-        "--tb=no",
+        "--tb=short",
         "-r a",
         "--log-cli-level=DEBUG",
         "--color=yes",
@@ -81,7 +81,7 @@ def main():
             f"--model={model}",
             f"--platform={rvc_platform}",
             "-v",
-            "--tb=no",
+            "--tb=short",
             "-r a",
             "--log-cli-level=DEBUG",
             "--color=yes",
diff --git a/tests/end_to_end/test_e2e.py b/tests/end_to_end/test_e2e.py
index cd3722dc..580a22f3 100644
--- a/tests/end_to_end/test_e2e.py
+++ b/tests/end_to_end/test_e2e.py
@@ -88,7 +88,7 @@ def test_pipelines(IP: str, ip_platform: str, nn_archive_path, model):
         elif e.returncode == 6:
             pytest.skip(f"Can't connect to the device with IP/mxid: {IP}")
         elif e.returncode == 7:
-            pytest.skip(f"Couldn't find model {model} in the ZOO")
+            pytest.skip(f"Couldn't find model {model} for {ip_platform} in the ZOO")
         elif e.returncode == 8:
             pytest.skip(
                 f"The model {model} is not supported in this test. (small input size, grayscale image, etc.)"