From a57ed3f53942efc58d51ea23eb39557782cb6652 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sat, 2 Dec 2023 11:39:44 -0800
Subject: [PATCH] hybrid fixes
---
semantic_router/layer.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/semantic_router/layer.py b/semantic_router/layer.py
index 070c9c0b..3bb685cc 100644
--- a/semantic_router/layer.py
+++ b/semantic_router/layer.py
@@ -1,5 +1,6 @@
import numpy as np
from numpy.linalg import norm
+from tqdm.auto import tqdm
from semantic_router.encoders import (
BaseEncoder,
@@ -27,7 +28,7 @@ class DecisionLayer:
# if decisions list has been passed, we initialize index now
if decisions:
# initialize index now
- for decision in decisions:
+ for decision in tqdm(decisions):
self._add_decision(decision=decision)
def __call__(self, text: str) -> str | None:
@@ -134,7 +135,7 @@ class HybridDecisionLayer:
# if decisions list has been passed, we initialize index now
if decisions:
# initialize index now
- for decision in decisions:
+ for decision in tqdm(decisions):
self._add_decision(decision=decision)
def __call__(self, text: str) -> str | None:
--
GitLab