From a34cc770c5c064599e1bf729f43639b93a1a2455 Mon Sep 17 00:00:00 2001 From: QianZhu Date: Tue, 2 Apr 2024 13:12:22 -0700 Subject: [PATCH] remote count_rows need to return the number (#1181) --- python/python/lancedb/remote/table.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python/python/lancedb/remote/table.py b/python/python/lancedb/remote/table.py index c1428a0b..d17410f8 100644 --- a/python/python/lancedb/remote/table.py +++ b/python/python/lancedb/remote/table.py @@ -500,7 +500,10 @@ class RemoteTable(Table): def count_rows(self, filter: Optional[str] = None) -> int: payload = {"predicate": filter} - self._conn._client.post(f"/v1/table/{self._name}/count_rows/", data=payload) + resp = self._conn._client.post( + f"/v1/table/{self._name}/count_rows/", data=payload + ) + return resp def add_columns(self, transforms: Dict[str, str]): raise NotImplementedError(