Skip to content
Snippets Groups Projects
Commit 95f88bb0 authored by Ismail Ashraq's avatar Ismail Ashraq
Browse files

remove print statements

parent d6397194
No related branches found
No related tags found
No related merge requests found
......@@ -68,7 +68,6 @@ class LocalIndex(BaseIndex):
if self.index is None or self.routes is None:
raise ValueError("Index or routes are not populated.")
if route_filter is not None:
print(f"Filtering routes with filter: {route_filter}")
filtered_index = []
filtered_routes = []
for route, vec in zip(self.routes, self.index):
......@@ -84,7 +83,6 @@ class LocalIndex(BaseIndex):
sim = similarity_matrix(vector, self.index)
scores, idx = top_scores(sim, top_k)
route_names = [self.routes[i] for i in idx]
print(f"Routes considered for similarity calculation: {route_names}")
return scores, route_names
def delete(self, route_name: str):
......
......@@ -229,7 +229,6 @@ class PineconeIndex(BaseIndex):
raise ValueError("Index is not populated.")
query_vector_list = vector.tolist()
if route_filter is not None:
print(f"Filtering routes with filter: {route_filter}")
filter_query = {"sr_route": {"$in": route_filter}}
else:
filter_query = None
......@@ -241,7 +240,6 @@ class PineconeIndex(BaseIndex):
)
scores = [result["score"] for result in results["matches"]]
route_names = [result["metadata"]["sr_route"] for result in results["matches"]]
print(f"Routes considered for similarity calculation: {route_names}")
return np.array(scores), route_names
def delete_index(self):
......
......@@ -241,7 +241,6 @@ class RouteLayer:
vector = self._encode(text=text)
route, top_class_scores = self._retrieve_top_route(vector, route_filter)
print(f"Selected route: {route.name if route else 'None'}")
passed = self._check_threshold(top_class_scores, route)
if passed and route is not None and not simulate_static:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment