.

harvardnlp · Sep 3, 2019 · 9ec9f52 · 9ec9f52
1 parent c4a6f72
commit 9ec9f52
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch_struct/data/data.py b/torch_struct/data/data.py
@@ -1,5 +1,6 @@
 
 import torchtext
+import torch
 
 def token_pre(tokenizer, q):
     st = " ".join(q)