fix kim data reader

BamLubi · Jul 12, 2022 · 7b6d0dc · 7b6d0dc
1 parent 0f1f4da
commit 7b6d0dc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/models/match/kim/mind_reader.py b/models/match/kim/mind_reader.py
@@ -13,6 +13,8 @@
 # limitations under the License.
 from datetime import datetime
 import time
+import nltk
+nltk.download('punkt')
 from nltk.tokenize import word_tokenize
 import numpy as np
 import os