Skip to content

Commit d60ff38

Browse files
committed
minor cleanup
1 parent 0b1e566 commit d60ff38

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

synthpop/processor/processor.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ def preprocess(self, df, dtypes):
8383
df.loc[:,col_nan_name] = df[col_nan_name].astype('category')
8484
self.spop.df_dtypes[col_nan_name] = 'category'
8585

86-
return df #Sex_NaN is a part of this data frame
86+
return df
8787

8888
def postprocess(self, synth_df):
8989
#sex_NaN is not a column of synth_df
@@ -113,7 +113,6 @@ def postprocess(self, synth_df):
113113

114114
# NaNs in numerical columns
115115
#The code below sets changes NANs in numerical columns to a given value, and removes the NAN indicator column.
116-
#The NAN_indicator columns are not synthesised.
117116
elif processing_nan_col_dict['dtype'] in NUM_COLS_DTYPES:
118117
for col_nan_flag, col_nan_value in processing_nan_col_dict['nan_flags'].items():
119118
nan_flag_indices = synth_df[processing_nan_col_dict['col_nan_name']] == col_nan_flag #expects columnName_NAN in the synthetic data set

0 commit comments

Comments
 (0)