diff --git a/semantic_router/layer.py b/semantic_router/layer.py
index 511540a39f176fc02f806c6693139a8e32e2c70f..e7381df0fd90d7ef8d909dbb146dc62c354837a9 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)