Fix bug in KBinsDiscretizer.set_attributes_from_dict

MatthiasRoelsPython · MatthiasRoelsPython · commit 8d8d5534b04f · 2020-04-21T14:05:26.000+02:00
diff --git a/cobra/preprocessing/kbins_discretizer.py b/cobra/preprocessing/kbins_discretizer.py
@@ -161,7 +161,7 @@ def set_attributes_from_dict(self, params: dict):
         self.set_params(**params)
 
         self._bins_by_column = {
-            key: [tuple(l) for l in value]
+            key: ([tuple(l) for l in value] if value else None)
             for key, value in _bins_by_column.items()
         }
 

Original file line number	Diff line number	Diff line change
`@@ -161,7 +161,7 @@ def set_attributes_from_dict(self, params: dict):`
`161`	`161`	`self.set_params(**params)`
`162`	`162`
`163`	`163`	`self._bins_by_column = {`
`164`		`- key: [tuple(l) for l in value]`
	`164`	`+ key: ([tuple(l) for l in value] if value else None)`
`165`	`165`	`for key, value in _bins_by_column.items()`
`166`	`166`	`}`
`167`	`167`