FEATURE/#364

mabel-dev · Aug 27, 2022 · 2fb3a38 · 2fb3a38
1 parent 1dcca00
commit 2fb3a38
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/opteryx/engine/functions/string_functions.py b/opteryx/engine/functions/string_functions.py
@@ -12,9 +12,6 @@
 
 import numpy
 
-from opteryx.third_party.soundex import Soundex
-
-
 def string_slicer_left(arr, length):
     """
     Slice a list of strings from the left
@@ -54,6 +51,8 @@ def string_slicer_right(arr, length):
 
 
 def soundex(arr):
+    from opteryx.third_party.fuzzy.soundex import Soundex
+
     _soundex = Soundex(4)
     interim = ["0000"] * arr.size
 

diff --git a/opteryx/third_party/fuzzy/__init__.py b/opteryx/third_party/fuzzy/__init__.py
diff --git a/opteryx/third_party/soundex.pyx → opteryx/third_party/fuzzy/csoundex.pyx b/opteryx/third_party/soundex.pyx → opteryx/third_party/fuzzy/csoundex.pyx
diff --git a/opteryx/third_party/fuzzy/soundex.py b/opteryx/third_party/fuzzy/soundex.py
@@ -0,0 +1 @@
+from csoundex import Soundex
diff --git a/setup.py b/setup.py
@@ -19,8 +19,8 @@
         include_dirs=[np.get_include()],
     ),
     Extension(
-        name="soundex",
-        sources=["opteryx/third_party/soundex.pyx"],
+        name="csoundex",
+        sources=["opteryx/third_party/fuzzy/csoundex.pyx"],
         include_dirs=[np.get_include()],
     ),
 ]

diff --git a/tests/misc/test_soundex.py b/tests/misc/test_soundex.py
@@ -4,7 +4,7 @@
 sys.path.insert(1, os.path.join(sys.path[0], "../.."))
 
 import pytest
-from opteryx.third_party.soundex import Soundex
+from opteryx.third_party.fuzzy.soundex import Soundex
 
 # fmt:off
 TESTS = [