Fix imports

2 files changed, 1 insertions, 2 deletions

diff --git a/WMD_matching.py b/WMD_matching.py
index 8581ffe..2316a10 100644
--- a/WMD_matching.py
+++ b/WMD_matching.py
@@ -1,7 +1,5 @@
 import argparse
 import numpy as np
-from mosestokenizer import *
-import nltk
 import random
 from sklearn.feature_extraction.text import CountVectorizer, TfidfVectorizer
 from sklearn.preprocessing import normalize
diff --git a/Wasserstein_Distance.py b/Wasserstein_Distance.py
index d8d08b8..86204a1 100644
--- a/Wasserstein_Distance.py
+++ b/Wasserstein_Distance.py
@@ -9,6 +9,7 @@ from sklearn.metrics.scorer import check_scoring
 from pathos.multiprocessing import ProcessingPool as Pool
 from sklearn.metrics import euclidean_distances
 import numpy as np
+from mosestokenizer import MosesDetokenizer
 
 class Wasserstein_Matcher(KNeighborsClassifier):
     """