only import numpy locally

rapidfuzz · Oct 6, 2023 · 11ed025 · 11ed025
1 parent a58bbf5
commit 11ed025
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_hypothesis.py b/tests/test_hypothesis.py
@@ -3,7 +3,6 @@
 from string import ascii_letters, digits, punctuation
 
 import hypothesis.strategies as st
-import numpy as np
 import pytest
 from hypothesis import assume, given, settings
 
@@ -168,6 +167,7 @@ def partial_ratio_short_needle(s1, s2):
 
 
 def cdist_scorer(queries, choices, scorer):
+    import numpy as np
     matrix = np.zeros((len(queries), len(choices)), dtype=np.uint8)
 
     for i, query in enumerate(queries):
@@ -178,6 +178,7 @@ def cdist_scorer(queries, choices, scorer):
 
 
 def cdist_distance(queries, choices, scorer):
+    import numpy as np
     matrix = np.zeros((len(queries), len(choices)), dtype=np.int32)
 
     for i, query in enumerate(queries):