remove cosine_distance/euclidean_distance impl

src/datachain/sql/sqlite/base.py

@@ -217,23 +217,6 @@ def register_user_defined_sql_functions() -> None:
     # Register optional functions if we have the necessary dependencies
     # and otherwise register functions that will raise an exception with
     # installation instructions
-    try:
-        from .vector import cosine_distance, euclidean_distance
-    except ImportError as exc:
-        # We want to throw an exception when trying to compile these
-        # functions and also if the functions are called using raw SQL.
-        cosine_distance = missing_vector_function("cosine_distance", exc)
-        euclidean_distance = missing_vector_function("euclidean_distance", exc)
-        _compiler_hooks["cosine_distance"] = cosine_distance
-        _compiler_hooks["euclidean_distance"] = euclidean_distance
-
-    def create_vector_functions(conn):
-        conn.create_function("cosine_distance", 2, cosine_distance, deterministic=True)
-        conn.create_function(
-            "euclidean_distance", 2, euclidean_distance, deterministic=True
-        )
-
-    _registered_function_creators["vector_functions"] = create_vector_functions
 
     def create_numeric_functions(conn):
         conn.create_function("divide", 2, lambda a, b: a / b, deterministic=True)

tests/unit/sql/test_array.py

@@ -1,6 +1,7 @@
 import math
 
 import pytest
+from pytest import approx
 
 from datachain import func
 from datachain.sql import select
@@ -14,7 +15,8 @@ def test_cosine_distance(warehouse):
         func.cosine_distance([0.0, 10.0], [1.0, 0.0]).label("cos4"),
     )
     result = tuple(warehouse.db.execute(query))
-    assert result == ((0.0, 0.0, 1.0, 1.0),)
+    # using approx due to https://github.com/unum-cloud/usearch/issues/320
+    assert result == ((approx(0.0, abs=1e-6), approx(0.0, abs=1e-6), 1.0, 1.0),)
 
 
 def test_euclidean_distance(warehouse):