From 60ce86211392405f2a8e8a575ea5ad80cb7d0107 Mon Sep 17 00:00:00 2001
From: maxbachmann <ssh@maxbachmann.de>
Date: Mon, 14 Sep 2020 23:55:34 +0200
Subject: [PATCH] use difflib for string matching

---
 TTS/bin/convert_melgan_torch_to_tf.py    | 4 ++--
 TTS/bin/convert_tacotron2_torch_to_tf.py | 4 ++--
 requirements.txt                         | 1 -
 3 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/TTS/bin/convert_melgan_torch_to_tf.py b/TTS/bin/convert_melgan_torch_to_tf.py
index a006b7b2..2eec6157 100644
--- a/TTS/bin/convert_melgan_torch_to_tf.py
+++ b/TTS/bin/convert_melgan_torch_to_tf.py
@@ -1,10 +1,10 @@
 import argparse
+from difflib import SequenceMatcher
 import os
 
 import numpy as np
 import tensorflow as tf
 import torch
-from fuzzywuzzy import fuzz
 
 from TTS.utils.io import load_config
 from TTS.vocoder.tf.utils.convert_torch_to_tf_utils import (
@@ -67,7 +67,7 @@ for tf_name in tf_var_names:
         continue
     tf_name_edited = convert_tf_name(tf_name)
     ratios = [
-        fuzz.ratio(torch_name, tf_name_edited)
+        SequenceMatcher(None, torch_name, tf_name_edited).ratio()
         for torch_name in torch_var_names
     ]
     max_idx = np.argmax(ratios)
diff --git a/TTS/bin/convert_tacotron2_torch_to_tf.py b/TTS/bin/convert_tacotron2_torch_to_tf.py
index 2ba1aa16..485e56be 100644
--- a/TTS/bin/convert_tacotron2_torch_to_tf.py
+++ b/TTS/bin/convert_tacotron2_torch_to_tf.py
@@ -1,6 +1,7 @@
 # %%
 # %%
 import argparse
+from difflib import SequenceMatcher
 import os
 import sys
 # %%
@@ -10,7 +11,6 @@ from pprint import pprint
 import numpy as np
 import tensorflow as tf
 import torch
-from fuzzywuzzy import fuzz
 from TTS.tts.tf.models.tacotron2 import Tacotron2
 from TTS.tts.tf.utils.convert_torch_to_tf_utils import (
     compare_torch_tf, convert_tf_name, transfer_weights_torch_to_tf)
@@ -106,7 +106,7 @@ for tf_name in tf_var_names:
         continue
     tf_name_edited = convert_tf_name(tf_name)
     ratios = [
-        fuzz.ratio(torch_name, tf_name_edited)
+        SequenceMatcher(None, torch_name, tf_name_edited).ratio()
         for torch_name in torch_var_names
     ]
     max_idx = np.argmax(ratios)
diff --git a/requirements.txt b/requirements.txt
index fdec4c57..f0f2c057 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -20,5 +20,4 @@ soundfile
 nose==1.3.7
 cardboardlint==1.3.0
 pylint==2.5.3
-fuzzywuzzy
 gdown