Bugfix for issue #56: removing just keys that exist in dataset when w…

…hen replacing text feature names concatenating their level
ludwig-ai · Feb 12, 2019 · 1820ef5 · 1820ef5
1 parent a290336
commit 1820ef5
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/ludwig/data/preprocessing.py b/ludwig/data/preprocessing.py
@@ -20,7 +20,6 @@
 
 import h5py
 import numpy as np
-import pandas as pd
 import yaml
 
 from ludwig.constants import *
@@ -771,8 +770,8 @@ def replace_text_feature_level(model_definition, datasets):
                     )
                 ]
                 for level in ('word', 'char'):
-                    del dataset[
-                        '{}_{}'.format(
+                    name_level = '{}_{}'.format(
                             feature['name'],
                             level)
-                    ]
+                    if name_level in dataset:
+                        del dataset[name_level]