bug fix with explore policy

This commit is contained in:
Steve Nyemba 2019-03-05 15:45:51 -06:00
parent 3ff382ad71
commit 30fe9cf9fc
1 changed files with 1 additions and 1 deletions

View File

@ -76,7 +76,7 @@ class deid :
sample = args['sample'] if 'sample' in args else pd.DataFrame(self._df) sample = args['sample'] if 'sample' in args else pd.DataFrame(self._df)
k = sample.columns.size -1 if 'field_count' not in args else int(args['field_count']) k = sample.columns.size if 'field_count' not in args else int(args['field_count']) + 1
if 'id' in args : if 'id' in args :
id = args['id'] id = args['id']
columns = list(set(sample.columns.tolist()) - set([id])) columns = list(set(sample.columns.tolist()) - set([id]))