prediction case bugfix
@@ -1126,7 +1126,7 @@ class NLUPipeline(BasePipe):
print(
'Could not find column named "text" in input Pandas Dataframe. Please ensure one column named such exists. Columns in DF are : ',
data.columns)
- if isinstance(data,pd.DataFrame): # casting follows pd->spark->pd
+ elif isinstance(data,pd.DataFrame): # casting follows pd->spark->pd
self.output_datatype = 'pandas'
# set first col as text column if there is none
