aboutsummaryrefslogtreecommitdiff
path: root/lstm_chem/generator.py
diff options
context:
space:
mode:
Diffstat (limited to 'lstm_chem/generator.py')
-rw-r--r--[-rwxr-xr-x]lstm_chem/generator.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/lstm_chem/generator.py b/lstm_chem/generator.py
index 498f864..4f80e9f 100755..100644
--- a/lstm_chem/generator.py
+++ b/lstm_chem/generator.py
@@ -1,6 +1,6 @@
from tqdm import tqdm
import numpy as np
-from lstm_chem.utils.smiles_tokenizer import SmilesTokenizer
+from lstm_chem.utils.smiles_tokenizer2 import SmilesTokenizer
class LSTMChemGenerator(object):