From 9912522985ad9548c5ca7670e408932de9f303d7 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Tue, 7 Jan 2025 08:46:18 +0400
Subject: [PATCH] fix: try-except logic
---
tests/unit/test_router.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py
index f3a7c6c7..90c6d020 100644
--- a/tests/unit/test_router.py
+++ b/tests/unit/test_router.py
@@ -750,8 +750,8 @@ class TestSemanticRouter:
count = 0
# we allow for 5 retries to allow for index to be populated
while count < RETRY_COUNT:
- query_result = route_layer(text="Hello").name
try:
+ query_result = route_layer(text="Hello").name
assert query_result in ["Route 1", "Route 2"]
break
except Exception:
@@ -787,8 +787,8 @@ class TestSemanticRouter:
count = 0
# we allow for 5 retries to allow for index to be populated
while count < RETRY_COUNT:
- query_result = route_layer(text="Hello", route_filter=["Route 1"]).name
try:
+ query_result = route_layer(text="Hello", route_filter=["Route 1"]).name
assert query_result in ["Route 1"]
break
except Exception:
--
GitLab