diff --git a/deeplake/client/managed/managed_client.py b/deeplake/client/managed/managed_client.py index 18bac2ae0d..cf4ecb97da 100644 --- a/deeplake/client/managed/managed_client.py +++ b/deeplake/client/managed/managed_client.py @@ -146,7 +146,7 @@ def vectorstore_add( def vectorstore_remove_rows( self, path: str, - indices: Optional[List[int]] = None, + row_ids: Optional[List[int]] = None, ids: Optional[List[str]] = None, filter: Optional[Dict[str, str]] = None, query: Optional[str] = None, @@ -157,7 +157,7 @@ def vectorstore_remove_rows( relative_url=VECTORSTORE_REMOVE_ROWS_SUFFIX, json={ "dataset": path, - "indices": indices, + "row_ids": row_ids, "ids": ids, "filter": filter, "query": query, diff --git a/deeplake/core/vectorstore/managed_vectorstore.py b/deeplake/core/vectorstore/managed_vectorstore.py index 0fbb330c7b..f1d8d11604 100644 --- a/deeplake/core/vectorstore/managed_vectorstore.py +++ b/deeplake/core/vectorstore/managed_vectorstore.py @@ -458,7 +458,7 @@ def delete( self.client.vectorstore_remove_rows( path=self.path, - indices=row_ids, + row_ids=row_ids, ids=ids, filter=filter, query=query,