fix: use backend to determine annlite DocumentArray length (docarray#327)

davidbp · web-flow · commit bbac194cd1e6 · 2022-05-05T16:28:08.000+02:00
* fix: len method annlite

* fix: add again annlite in setup
diff --git a/docarray/array/storage/annlite/backend.py b/docarray/array/storage/annlite/backend.py
@@ -98,3 +98,6 @@ def __setstate__(self, state):
         from annlite import AnnLite
 
         self._annlite = AnnLite(n_dim, lock=False, **filter_dict(config))
+
+    def __len__(self):
+        return self._annlite.index_size
diff --git a/setup.py b/setup.py
@@ -71,6 +71,9 @@
         'qdrant': [
             'qdrant-client~=0.7.0',
         ],
+        'annlite': [
+            'annlite>=0.3.0',
+        ],
         'test': [
             'pytest',
             'pytest-timeout',
