Merge pull request #1319 from maxbachmann-forks/master

use rapidfuzz instead of fuzzywuzzy

Author: willingc

nbgrader/converters/base.py

@@ -5,6 +5,7 @@
 import sqlalchemy
 import traceback
 
+from rapidfuzz import fuzz
 from traitlets.config import LoggingConfigurable, Config
 from traitlets import Bool, List, Dict, Integer, Instance, Type
 from traitlets import default
@@ -120,11 +121,6 @@ def init_notebooks(self) -> None:
             assignment_glob2 = self._format_source("*", self.coursedir.student_id)
             found = glob.glob(assignment_glob2)
             if found:
-                # Normally it is a bad idea to put imports in the middle of
-                # a function, but we do this here because otherwise fuzzywuzzy
-                # prints an annoying message about python-Levenshtein every
-                # time nbgrader is run.
-                from fuzzywuzzy import fuzz
                 scores = sorted([(fuzz.ratio(assignment_glob, x), x) for x in found])
                 self.log.error("Did you mean: %s", scores[-1][1])
 

nbgrader/exchange/exchange.py

@@ -8,6 +8,7 @@
 
 from dateutil.tz import gettz
 from dateutil.parser import parse
+from rapidfuzz import fuzz
 from traitlets.config import LoggingConfigurable
 from traitlets import Unicode, Bool, Instance, Type, default, validate, TraitError
 from jupyter_core.paths import jupyter_data_dir
@@ -177,11 +178,6 @@ def _assignment_not_found(self, src_path, other_path):
         self.log.fatal(msg)
         found = glob.glob(other_path)
         if found:
-            # Normally it is a bad idea to put imports in the middle of
-            # a function, but we do this here because otherwise fuzzywuzzy
-            # prints an annoying message about python-Levenshtein every
-            # time nbgrader is run.
-            from fuzzywuzzy import fuzz
             scores = sorted([(fuzz.ratio(self.src_path, x), x) for x in found])
             self.log.error("Did you mean: %s", scores[-1][1])
 

setup.py

@@ -100,7 +100,7 @@
         "requests",
         "jsonschema",
         "alembic",
-        "fuzzywuzzy"
+        "rapidfuzz"
     ]
 )