kmaurinjones commited on
Commit
a7bfbfb
1 Parent(s): b26dd30

Update songscope.py

Browse files
Files changed (1) hide show
  1. songscope.py +6 -3
songscope.py CHANGED
@@ -118,7 +118,8 @@ def find_artist(artist_name: str):
118
  min_id = None
119
  max_sim = -100
120
  for id, name in enumerate(artist_urls):
121
- dist = Levenshtein.jaro(artist_name, name)
 
122
  if max_sim < dist:
123
  max_sim = dist
124
  min_id = id
@@ -243,7 +244,8 @@ def find_artist(artist_name: str) -> str:
243
  min_id = None
244
  max_sim = -100
245
  for id, name in enumerate(artist_urls):
246
- dist = Levenshtein.jaro(artist_name, name)
 
247
  if max_sim < dist:
248
  max_sim = dist
249
  min_id = id
@@ -448,7 +450,8 @@ def get_metadata(artist_name: str, song_titles: list = None) -> dict:
448
  min_id = None
449
  max_sim = -100
450
  for id, az_name in enumerate(az_titles):
451
- dist = Levenshtein.jaro(title, az_name)
 
452
  if max_sim < dist:
453
  max_sim = dist
454
  min_id = id
 
118
  min_id = None
119
  max_sim = -100
120
  for id, name in enumerate(artist_urls):
121
+ # dist = Levenshtein.jaro(artist_name, name)
122
+ dist = homemade_jaro(artist_name, name)
123
  if max_sim < dist:
124
  max_sim = dist
125
  min_id = id
 
244
  min_id = None
245
  max_sim = -100
246
  for id, name in enumerate(artist_urls):
247
+ # dist = Levenshtein.jaro(artist_name, name)
248
+ dist = homemade_jaro(artist_name, name)
249
  if max_sim < dist:
250
  max_sim = dist
251
  min_id = id
 
450
  min_id = None
451
  max_sim = -100
452
  for id, az_name in enumerate(az_titles):
453
+ # dist = Levenshtein.jaro(title, az_name)
454
+ dist = homemade_jaro(title, az_name)
455
  if max_sim < dist:
456
  max_sim = dist
457
  min_id = id