Skip to content
Snippets Groups Projects
Unverified Commit 985b7d10 authored by Bogdan Buduroiu's avatar Bogdan Buduroiu
Browse files

fix: base router encoding type

parent 83ee4923
No related branches found
No related tags found
No related merge requests found
......@@ -562,7 +562,7 @@ class BaseRouter(BaseModel):
if vector is None:
if text is None:
raise ValueError("Either text or vector must be provided")
vector = self._encode(text=[text])
vector = self._encode(text=[text], input_type="queries")
# convert to numpy array if not already
vector = xq_reshape(vector)
# get scores and routes
......@@ -635,7 +635,7 @@ class BaseRouter(BaseModel):
if vector is None:
if text is None:
raise ValueError("Either text or vector must be provided")
vector = await self._async_encode(text=[text])
vector = await self._async_encode(text=[text], input_type="queries")
# convert to numpy array if not already
vector = xq_reshape(vector)
# get scores and routes
......
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