leavoigt commited on
Commit
7bb152d
1 Parent(s): 5cb79de

Update utils/target_classifier.py

Browse files
Files changed (1) hide show
  1. utils/target_classifier.py +1 -1
utils/target_classifier.py CHANGED
@@ -105,4 +105,4 @@ def target_classification(haystack_doc:pd.DataFrame,
105
  # df.index += 1
106
  # # df['Label_def'] = df['Target Label'].apply(lambda i: _lab_dict[i])
107
 
108
- return df
 
105
  # df.index += 1
106
  # # df['Label_def'] = df['Target Label'].apply(lambda i: _lab_dict[i])
107
 
108
+ return haystack_doc