Fix #166

brightway-lca · May 14, 2024 · 9e3e613 · 9e3e613
1 parent 66dab24
commit 9e3e613
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 1 deletion.
diff --git a/bw2data/backends/typos.py b/bw2data/backends/typos.py
@@ -1,7 +1,14 @@
 from typing import Iterable
 import warnings
 from functools import partial
-from ..string_distance import damerau_levenshtein
+
+try:
+    from rapidfuzz.distance import DamerauLevenshtein
+    damerau_levenshtein = DamerauLevenshtein.distance
+except ImportError:
+    # Can happen on Windows, see
+    # https://github.com/rapidfuzz/RapidFuzz/tree/main?tab=readme-ov-file#with-pip
+    from ..string_distance import damerau_levenshtein
 
 VALID_ACTIVITY_TYPES = (
     "process",

diff --git a/requirements-test.txt b/requirements-test.txt
@@ -8,6 +8,7 @@ platformdirs
 pytest
 pytest-cov
 python-coveralls
+rapidfuzz
 requests>=1.1.0
 scipy
 stats_arrays

diff --git a/requirements.txt b/requirements.txt
@@ -5,6 +5,7 @@ numpy<2
 peewee>=3.9.4
 platformdirs
 requests>=1.1.0
+rapidfuzz
 scipy
 stats_arrays
 tqdm

diff --git a/setup.py b/setup.py
@@ -34,6 +34,7 @@
         "numpy<2",
         "peewee>=3.9.4",
         "platformdirs",
+        "rapidfuzz",
         "scipy",
         "stats_arrays",
         "tqdm",