From c6ffb6cff5e6394821c9778a38cc5410ecdb9673 Mon Sep 17 00:00:00 2001
From: Bogdan Buduroiu <bogdan@buduroiu.com>
Date: Tue, 28 Nov 2023 10:49:10 +0800
Subject: [PATCH] Black formatting
---
semantic_router/layer.py | 8 ++++++--
tests/functional/test_linear.py | 2 +-
2 files changed, 7 insertions(+), 3 deletions(-)
diff --git a/semantic_router/layer.py b/semantic_router/layer.py
index dd746d0e..fcedc979 100644
--- a/semantic_router/layer.py
+++ b/semantic_router/layer.py
@@ -69,7 +69,9 @@ class DecisionLayer:
scores, idx = top_scores(sim, top_k)
# get the utterance categories (decision names)
decisions = self.categories[idx] if self.categories is not None else []
- return [{"decision": d, "score": s.item()} for d, s in zip(decisions, scores)]
+ return [
+ {"decision": d, "score": s.item()} for d, s in zip(decisions, scores)
+ ]
else:
return []
@@ -84,7 +86,9 @@ class DecisionLayer:
scores_by_class[decision] = [score]
# Calculate total score for each class
- total_scores = {decision: sum(scores) for decision, scores in scores_by_class.items()}
+ total_scores = {
+ decision: sum(scores) for decision, 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/functional/test_linear.py b/tests/functional/test_linear.py
index 6771fd9c..210de6d2 100644
--- a/tests/functional/test_linear.py
+++ b/tests/functional/test_linear.py
@@ -1,5 +1,5 @@
-import pytest
import numpy as np
+import pytest
from semantic_router.linear import similarity_matrix, top_scores
--
GitLab