Skip to content
Snippets Groups Projects
Commit ea09babc authored by zahid-syed's avatar zahid-syed
Browse files

fixed linting

parent 231fc6ec
No related branches found
No related tags found
No related merge requests found
...@@ -435,7 +435,7 @@ class RouteLayer: ...@@ -435,7 +435,7 @@ class RouteLayer:
max_iter: int = 500, max_iter: int = 500,
): ):
# convert inputs into array # convert inputs into array
Xq = [] Xq: List[List[float]] = []
for i in tqdm(range(0, len(X), batch_size), desc="Processing batches"): for i in tqdm(range(0, len(X), batch_size), desc="Processing batches"):
emb = np.array(self.encoder(X[i : i + batch_size])) emb = np.array(self.encoder(X[i : i + batch_size]))
Xq.extend(emb) Xq.extend(emb)
...@@ -465,7 +465,7 @@ class RouteLayer: ...@@ -465,7 +465,7 @@ class RouteLayer:
""" """
Evaluate the accuracy of the route selection. 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"): for i in tqdm(range(0, len(X), batch_size), desc="Processing batches"):
emb = np.array(self.encoder(X[i : i + batch_size])) emb = np.array(self.encoder(X[i : i + batch_size]))
Xq.extend(emb) Xq.extend(emb)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment