Skip to content
Snippets Groups Projects
Unverified Commit a57ed3f5 authored by James Briggs's avatar James Briggs
Browse files

hybrid fixes

parent b3fe444a
No related branches found
No related tags found
No related merge requests found
import numpy as np import numpy as np
from numpy.linalg import norm from numpy.linalg import norm
from tqdm.auto import tqdm
from semantic_router.encoders import ( from semantic_router.encoders import (
BaseEncoder, BaseEncoder,
...@@ -27,7 +28,7 @@ class DecisionLayer: ...@@ -27,7 +28,7 @@ class DecisionLayer:
# if decisions list has been passed, we initialize index now # if decisions list has been passed, we initialize index now
if decisions: if decisions:
# initialize index now # initialize index now
for decision in decisions: for decision in tqdm(decisions):
self._add_decision(decision=decision) self._add_decision(decision=decision)
def __call__(self, text: str) -> str | None: def __call__(self, text: str) -> str | None:
...@@ -134,7 +135,7 @@ class HybridDecisionLayer: ...@@ -134,7 +135,7 @@ class HybridDecisionLayer:
# if decisions list has been passed, we initialize index now # if decisions list has been passed, we initialize index now
if decisions: if decisions:
# initialize index now # initialize index now
for decision in decisions: for decision in tqdm(decisions):
self._add_decision(decision=decision) self._add_decision(decision=decision)
def __call__(self, text: str) -> str | None: def __call__(self, text: str) -> str | None:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment