Fix bug on pair

emalgorithm · emalgorithm · commit 6065b4c0bffc · 2019-03-06T16:24:06.000Z
diff --git a/train.py b/train.py
@@ -19,7 +19,7 @@ def main():
 
     if opt.graph:
         lang, pairs = prepare_data(num_samples=opt.n_samples)
-        pairs = [pair for pair in pairs if len(pair[0][1]) > 0]
+        pairs = [pair for pair in pairs if len(pair[0][1][0]) > 0]
     else:
         lang, pairs = prepare_tokens(num_samples=opt.n_samples)
 
