aboutsummaryrefslogtreecommitdiff
path: root/lstm_chem/data_loader.py
diff options
context:
space:
mode:
Diffstat (limited to 'lstm_chem/data_loader.py')
-rw-r--r--[-rwxr-xr-x]lstm_chem/data_loader.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/lstm_chem/data_loader.py b/lstm_chem/data_loader.py
index 86ddbba..243f7e6 100755..100644
--- a/lstm_chem/data_loader.py
+++ b/lstm_chem/data_loader.py
@@ -3,7 +3,7 @@ import os
import numpy as np
from tqdm import tqdm
from tensorflow.keras.utils import Sequence
-from lstm_chem.utils.smiles_tokenizer import SmilesTokenizer
+from lstm_chem.utils.smiles_tokenizer2 import SmilesTokenizer
class DataLoader(Sequence):