From 231fc6ec49427407a9d043cf1bf682dbfc18c048 Mon Sep 17 00:00:00 2001
From: zahid-syed <zahid.s2618@gmail.com>
Date: Mon, 19 Feb 2024 20:00:42 -0500
Subject: [PATCH] fixed linting
---
semantic_router/layer.py | 13 ++++++-------
1 file changed, 6 insertions(+), 7 deletions(-)
diff --git a/semantic_router/layer.py b/semantic_router/layer.py
index 02f07b6b..511540a3 100644
--- a/semantic_router/layer.py
+++ b/semantic_router/layer.py
@@ -431,14 +431,13 @@ class RouteLayer:
self,
X: List[str],
y: List[str],
- batch_size : int,
+ batch_size: int,
max_iter: int = 500,
-
):
# convert inputs into array
Xq = []
- for i in tqdm(range(0, len(X), batch_size), desc= "Processing batches"):
- emb = np.array(self.encoder(X[i:i+batch_size]))
+ for i in tqdm(range(0, len(X), batch_size), desc="Processing batches"):
+ emb = np.array(self.encoder(X[i : i + batch_size]))
Xq.extend(emb)
# initial eval (we will iterate from here)
best_acc = self._vec_evaluate(Xq=np.array(Xq), y=y)
@@ -467,9 +466,9 @@ class RouteLayer:
Evaluate the accuracy of the route selection.
"""
Xq = []
- for i in tqdm(range(0,len(X),batch_size), desc="Processing batches"):
- emb = np.array(self.encoder(X[i:i+batch_size]))
- Xq.extend(emb)
+ for i in tqdm(range(0, len(X), batch_size), desc="Processing batches"):
+ emb = np.array(self.encoder(X[i : i + batch_size]))
+ Xq.extend(emb)
accuracy = self._vec_evaluate(Xq=np.array(Xq), y=y)
return accuracy
--
GitLab