From 61c407d9740e20c5a35ec92ea008064559373428 Mon Sep 17 00:00:00 2001 From: Ash Vardanian <1983160+ashvardanian@users.noreply.github.com> Date: Mon, 19 Feb 2024 14:55:08 -0800 Subject: [PATCH] Move StringZilla import into the function --- nltk/metrics/distance.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/nltk/metrics/distance.py b/nltk/metrics/distance.py index 5addcae1fe..08ee986510 100644 --- a/nltk/metrics/distance.py +++ b/nltk/metrics/distance.py @@ -22,8 +22,6 @@ import operator import warnings -import stringzilla as sz - def _edit_dist_init(len1, len2): lev = [] @@ -91,7 +89,9 @@ def edit_distance(s1, s2, substitution_cost=1, transpositions=False): # StringZilla currently doesn't support transpositions and cost-parameterized substitutions. # It's C implementation is 10x faster than a pure Python implementation. if not transpositions and substitution_cost == 1: - return sz.edit_distance_unicode(s1, s2) + from stringzilla import edit_distance_unicode as sz_edit_dist + + return sz_edit_dist(s1, s2) # set up a 2-D array len1 = len(s1)