From 5aa6c77e8772f8611f3a70d82c9101f8fb01e9d6 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Tue, 12 Dec 2023 12:47:09 -0800
Subject: [PATCH] lint
---
semantic_router/layer.py | 16 ++++------------
tests/unit/test_layer.py | 16 ++++------------
tests/unit/test_schema.py | 4 +---
3 files changed, 9 insertions(+), 27 deletions(-)
diff --git a/semantic_router/layer.py b/semantic_router/layer.py
index c29f190f..23e3ec69 100644
--- a/semantic_router/layer.py
+++ b/semantic_router/layer.py
@@ -75,9 +75,7 @@ class RouteLayer:
scores, idx = top_scores(sim, top_k)
# get the utterance categories (route names)
routes = self.categories[idx] if self.categories is not None else []
- return [
- {"route": d, "score": s.item()} for d, s in zip(routes, scores)
- ]
+ return [{"route": d, "score": s.item()} for d, s in zip(routes, scores)]
else:
return []
@@ -92,9 +90,7 @@ class RouteLayer:
scores_by_class[route] = [score]
# Calculate total score for each class
- total_scores = {
- route: sum(scores) for route, scores in scores_by_class.items()
- }
+ total_scores = {route: sum(scores) for route, scores in scores_by_class.items()}
top_class = max(total_scores, key=lambda x: total_scores[x], default=None)
# Return the top class and its associated scores
@@ -201,9 +197,7 @@ class HybridRouteLayer:
scores = total_sim[idx]
# get the utterance categories (route names)
routes = self.categories[idx] if self.categories is not None else []
- return [
- {"route": d, "score": s.item()} for d, s in zip(routes, scores)
- ]
+ return [{"route": d, "score": s.item()} for d, s in zip(routes, scores)]
else:
return []
@@ -224,9 +218,7 @@ class HybridRouteLayer:
scores_by_class[route] = [score]
# Calculate total score for each class
- total_scores = {
- route: sum(scores) for route, scores in scores_by_class.items()
- }
+ total_scores = {route: sum(scores) for route, scores in scores_by_class.items()}
top_class = max(total_scores, key=lambda x: total_scores[x], default=None)
# Return the top class and its associated scores
diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index 0571071a..34ee8b99 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -117,9 +117,7 @@ class TestRouteLayer:
class TestHybridRouteLayer:
def test_initialization(self, openai_encoder, routes):
- route_layer = HybridRouteLayer(
- encoder=openai_encoder, routes=routes
- )
+ route_layer = HybridRouteLayer(encoder=openai_encoder, routes=routes)
assert route_layer.score_threshold == 0.82
assert len(route_layer.index) == 5
assert len(set(route_layer.categories)) == 2
@@ -146,9 +144,7 @@ class TestHybridRouteLayer:
assert len(set(route_layer.categories)) == 2
def test_query_and_classification(self, openai_encoder, routes):
- route_layer = HybridRouteLayer(
- encoder=openai_encoder, routes=routes
- )
+ route_layer = HybridRouteLayer(encoder=openai_encoder, routes=routes)
query_result = route_layer("Hello")
assert query_result in ["Route 1", "Route 2"]
@@ -157,9 +153,7 @@ class TestHybridRouteLayer:
assert route_layer("Anything") is None
def test_semantic_classify(self, openai_encoder, routes):
- route_layer = HybridRouteLayer(
- encoder=openai_encoder, routes=routes
- )
+ route_layer = HybridRouteLayer(encoder=openai_encoder, routes=routes)
classification, score = route_layer._semantic_classify(
[
{"route": "Route 1", "score": 0.9},
@@ -170,9 +164,7 @@ class TestHybridRouteLayer:
assert score == [0.9]
def test_semantic_classify_multiple_routes(self, openai_encoder, routes):
- route_layer = HybridRouteLayer(
- encoder=openai_encoder, routes=routes
- )
+ route_layer = HybridRouteLayer(encoder=openai_encoder, routes=routes)
classification, score = route_layer._semantic_classify(
[
{"route": "Route 1", "score": 0.9},
diff --git a/tests/unit/test_schema.py b/tests/unit/test_schema.py
index 9128835a..7c954805 100644
--- a/tests/unit/test_schema.py
+++ b/tests/unit/test_schema.py
@@ -49,9 +49,7 @@ class TestSemanticSpaceDataclass:
assert semantic_space.routes == []
def test_semanticspace_add_route(self):
- route = Route(
- name="test", utterances=["hello", "hi"], description="greeting"
- )
+ route = Route(name="test", utterances=["hello", "hi"], description="greeting")
semantic_space = SemanticSpace()
semantic_space.add(route)
--
GitLab