include simd binaries in pyinstaller builds

rapidfuzz · Jul 5, 2024 · 6bbd17e · 6bbd17e
1 parent 8af875c
commit 6bbd17e
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 14 deletions.
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -1,6 +1,12 @@
 Changelog
 ---------
 
+[3.9.5] -
+^^^^^^^^^^^^^^^^^^^^
+Fixed
+~~~~~
+* include simd binaries in pyinstaller builds
+
 [3.9.4] - 2024-07-02
 ^^^^^^^^^^^^^^^^^^^^
 Fixed

diff --git a/src/rapidfuzz/__pyinstaller/hook-rapidfuzz.py b/src/rapidfuzz/__pyinstaller/hook-rapidfuzz.py
@@ -1,17 +1,12 @@
 # Pyinstaller hook to successfully freeze: https://pyinstaller.readthedocs.io/en/stable/hooks.html
 from __future__ import annotations
 
-hiddenimports = [
-    "array",
-    "rapidfuzz.fuzz_py",
-    "rapidfuzz.fuzz_cpp",
-    "rapidfuzz.utils_py",
-    "rapidfuzz.utils_cpp",
-    "rapidfuzz.process_py",
-    "rapidfuzz.process_cpp",
-    # distances
-    "rapidfuzz.distance._initialize_py",
-    "rapidfuzz.distance._initialize_cpp",
-    "rapidfuzz.distance.metrics_cpp",
-    "rapidfuzz.distance.metrics_py",
-]
+from PyInstaller.utils.hooks import collect_submodules
+
+def filterUnneededImports(name):
+    if "__pyinstaller" in name:
+        return False
+
+    return True
+
+hiddenimports = collect_submodules("rapidfuzz", filter=filterUnneededImports)