use rapidfuzz instead of fuzzywuzzy

pull/1/head
maxbachmann 2020-03-23 02:16:36 +01:00
parent e8035863b8
commit 9388a9820f
No known key found for this signature in database
GPG Key ID: 60334E83C23820B8
3 changed files with 4 additions and 5 deletions

View File

@ -65,6 +65,6 @@ setup(
},
extras_require={
'taxii': ['taxii2-client'],
'semantic': ['haversine', 'fuzzywuzzy'],
'semantic': ['haversine', 'rapidfuzz'],
},
)

View File

@ -363,8 +363,8 @@ def partial_string_based(str1, str2):
float: Number between 0.0 and 1.0 depending on match criteria.
"""
from fuzzywuzzy import fuzz
result = fuzz.token_sort_ratio(str1, str2, force_ascii=False)
from rapidfuzz import fuzz
result = fuzz.token_sort_ratio(str1, str2)
logger.debug("--\t\tpartial_string_based '%s' '%s'\tresult: '%s'", str1, str2, result)
return result / 100.0

View File

@ -9,9 +9,8 @@ deps =
pytest-cov
coverage
taxii2-client
fuzzywuzzy
rapidfuzz
haversine
python-Levenshtein
medallion
commands =
python -m pytest --cov=stix2 stix2/test/ --cov-report term-missing -W ignore::stix2.exceptions.STIXDeprecationWarning