From 6bd5da3b0eba9d056bf6e8f9a947629ede956549 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E2=80=9CDaniel=20Griffiths=E2=80=9D?=
<Danielgriffiths1790@gmail.com>
Date: Fri, 5 Jan 2024 09:53:55 +0000
Subject: [PATCH] fix: black lint reformat hybrid layer
---
semantic_router/hybrid_layer.py | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/semantic_router/hybrid_layer.py b/semantic_router/hybrid_layer.py
index 853fb332..40c95d63 100644
--- a/semantic_router/hybrid_layer.py
+++ b/semantic_router/hybrid_layer.py
@@ -36,7 +36,9 @@ class HybridRouteLayer:
else:
self.score_threshold = 0.82
# if routes list has been passed, we initialize index now
- if isinstance(self.sparse_encoder, TfidfEncoder) and hasattr(self.sparse_encoder, 'fit'):
+ if isinstance(self.sparse_encoder, TfidfEncoder) and hasattr(
+ self.sparse_encoder, "fit"
+ ):
self.sparse_encoder.fit(routes)
if routes:
# initialize index now
@@ -54,7 +56,9 @@ class HybridRouteLayer:
return None
def add(self, route: Route):
- if isinstance(self.sparse_encoder, TfidfEncoder) and hasattr(self.sparse_encoder, 'fit'):
+ if isinstance(self.sparse_encoder, TfidfEncoder) and hasattr(
+ self.sparse_encoder, "fit"
+ ):
self.sparse_encoder.fit(self.routes + [route])
self.sparse_index = None
for r in self.routes:
--
GitLab