only import numpy locally

rapidfuzz · Sep 7, 2023 · cc3cc7e · cc3cc7e
1 parent 762624b
commit cc3cc7e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_hypothesis.py b/tests/test_hypothesis.py
@@ -5,7 +5,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
 
@@ -164,6 +163,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):
@@ -174,6 +174,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):