Merge pull request #698 from wearepal/nan-to-num

tmke8 · web-flow · commit f992eda388ec · 2022-06-28T12:12:14.000+02:00
Fix usage of nan_to_num
diff --git a/ethicml/data/tabular_data/acs.py b/ethicml/data/tabular_data/acs.py
@@ -486,7 +486,7 @@ def load(
             target_transform=lambda x: x == 1,
             group=self.split,
             preprocess=lambda x: x,
-            postprocess=lambda x: np.nan_to_num(x, -1),
+            postprocess=lambda x: np.nan_to_num(x, nan=-1),
         )
 
         dataframe = data_obj._preprocess(dataframe)

Original file line number	Diff line number	Diff line change
`@@ -486,7 +486,7 @@ def load(`
`486`	`486`	`target_transform=lambda x: x == 1,`
`487`	`487`	`group=self.split,`
`488`	`488`	`preprocess=lambda x: x,`
`489`		`- postprocess=lambda x: np.nan_to_num(x, -1),`
	`489`	`+ postprocess=lambda x: np.nan_to_num(x, nan=-1),`
`490`	`490`	`)`
`491`	`491`
`492`	`492`	`dataframe = data_obj._preprocess(dataframe)`