From c651b4afb8ced5b12b41b4100cde594b2fb57013 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Fri, 12 Jan 2024 00:24:14 +0000
Subject: [PATCH] lint
---
semantic_router/encoders/tfidf.py | 12 +++++++-----
semantic_router/hybrid_layer.py | 4 ++--
tests/unit/encoders/test_tfidf.py | 3 ++-
3 files changed, 11 insertions(+), 8 deletions(-)
diff --git a/semantic_router/encoders/tfidf.py b/semantic_router/encoders/tfidf.py
index 5e81a463..0809b5ad 100644
--- a/semantic_router/encoders/tfidf.py
+++ b/semantic_router/encoders/tfidf.py
@@ -1,11 +1,13 @@
-import numpy as np
-from collections import Counter
-from semantic_router.encoders import BaseEncoder
-from semantic_router.route import Route
-from numpy.linalg import norm
import string
+from collections import Counter
from typing import Dict
+
+import numpy as np
from numpy import ndarray
+from numpy.linalg import norm
+
+from semantic_router.encoders import BaseEncoder
+from semantic_router.route import Route
class TfidfEncoder(BaseEncoder):
diff --git a/semantic_router/hybrid_layer.py b/semantic_router/hybrid_layer.py
index b56e1cd5..62c87efc 100644
--- a/semantic_router/hybrid_layer.py
+++ b/semantic_router/hybrid_layer.py
@@ -27,13 +27,13 @@ class HybridRouteLayer:
):
self.encoder = encoder
self.score_threshold = self.encoder.score_threshold
-
+
if sparse_encoder is None:
logger.warning("No sparse_encoder provided. Using default BM25Encoder.")
self.sparse_encoder = BM25Encoder()
else:
self.sparse_encoder = sparse_encoder
-
+
self.alpha = alpha
self.routes = routes
if isinstance(self.sparse_encoder, TfidfEncoder) and hasattr(
diff --git a/tests/unit/encoders/test_tfidf.py b/tests/unit/encoders/test_tfidf.py
index 6bb9fab3..7664433d 100644
--- a/tests/unit/encoders/test_tfidf.py
+++ b/tests/unit/encoders/test_tfidf.py
@@ -1,7 +1,8 @@
+import numpy as np
import pytest
+
from semantic_router.encoders import TfidfEncoder
from semantic_router.route import Route
-import numpy as np
@pytest.fixture
--
GitLab