Merge pull request #571 from aperture-data/release-0.4.44

Release 0.4.44

Author: gsaluja9

aperturedb/Utils.py

@@ -473,7 +473,8 @@ def count_connections(self, connections_class, constraints=None) -> int:
 
         return total_connections
 
-    def add_descriptorset(self, name: str, dim: int, metric="L2", engine="FaissFlat",
+    def add_descriptorset(self, name: str, dim: int, metric=["CS"],
+                          engine=["HNSW"],
                           properties: Optional[Dict] = None) -> bool:
         """
         Add a descriptor set to the database.

aperturedb/__init__.py

@@ -8,15 +8,21 @@
 import platform
 import faulthandler
 import signal
+import sys
+
+__version__ = "0.4.44"
+
+logger = logging.getLogger(__name__)
 
 # https://docs.python.org/3/library/faulthandler.html
 # Register SIGUSR1 to dump the stack trace
 # Good for debugging a running process
-faulthandler.register(signal.SIGUSR1.value)
-
-logger = logging.getLogger(__name__)
 
-__version__ = "0.4.43"
+if os.getenv("ADB_DEBUGGABLE", None) != None:
+    if sys.platform == "win32":
+        logger.warn("Unable to configure debugging support for win32")
+    else:
+        faulthandler.register(signal.SIGUSR1.value)
 
 # set log level
 formatter = logging.Formatter(

aperturedb/transformers/clip_pytorch_embeddings.py

@@ -24,7 +24,8 @@ def __init__(self, data: Subscriptable, **kwargs) -> None:
         if len(self._add_image_index) > 0:
             sample = generate_embedding(self.data[0][1][0])
             utils = self.get_utils()
-            utils.add_descriptorset(self.search_set_name, dim=len(sample) // 4)
+            utils.add_descriptorset(
+                self.search_set_name, dim=len(sample) // 4, metric=["CS"])
 
     def getitem(self, subscript):
         x = self.data[subscript]