bugfix for import

text-machine-lab · Feb 12, 2018 · 5f33917 · 5f33917
1 parent d451778
commit 5f33917
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/code/predict.py b/code/predict.py
@@ -169,7 +169,7 @@ def predict(files, model_path, output_dir, format, use_lstm=True):
             #continue
         #'''
 
-        sys.stdout.write('%s\n' % '-' * 30)
+        sys.stdout.write('%s\n' % ('-' * 30))
         sys.stdout.write('\n\t%d of %d\n' % (i+1,n))
         sys.stdout.write('\t%s\n\n' % txt)
 

diff --git a/code/tools.py b/code/tools.py
@@ -10,6 +10,7 @@
 import os
 import errno
 import string
+import math
 import re
 import pickle
 import numpy as np