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 @@ def __init__( ): 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