fix SettingWithCopyWarning

Signed-off-by: Samuel Hoffman <hoffman.sc@gmail.com>
Trusted-AI · Dec 3, 2021 · da6d549 · da6d549
1 parent 591fb1c
commit da6d549
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/aif360/sklearn/datasets/meps_datasets.py b/aif360/sklearn/datasets/meps_datasets.py
@@ -117,7 +117,8 @@ def fetch_meps(panel, *, accept_terms=None, data_home=None, cache=True,
                                labels=['< 10 Visits', '>= 10 Visits'])#['low', 'high'])
 
     # TODO: let standardize_dataset handle dropna (see above todo re: extra cols)
-    return standardize_dataset(df.dropna(), prot_attr='RACE', target='UTILIZATION',
+    df = df.dropna()
+    return standardize_dataset(df, prot_attr='RACE', target='UTILIZATION',
                                sample_weight='PERWT', usecols=usecols,
                                dropcols=dropcols, numeric_only=numeric_only,
                                dropna=dropna)