From ea09babc1e553942c3934e49f14a2ccf9549d127 Mon Sep 17 00:00:00 2001
From: zahid-syed <zahid.s2618@gmail.com>
Date: Mon, 19 Feb 2024 20:11:30 -0500
Subject: [PATCH] fixed linting
---
semantic_router/layer.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/semantic_router/layer.py b/semantic_router/layer.py
index 511540a3..e7381df0 100644
--- a/semantic_router/layer.py
+++ b/semantic_router/layer.py
@@ -435,7 +435,7 @@ class RouteLayer:
max_iter: int = 500,
):
# convert inputs into array
- Xq = []
+ Xq: List[List[float]] = []
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)
@@ -465,7 +465,7 @@ class RouteLayer:
"""
Evaluate the accuracy of the route selection.
"""
- Xq = []
+ Xq: List[List[float]] = []
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)
--
GitLab