[ENH, BUGFIX] Add class_name arg to make_step and fix missing *args bug

alegonz / baikal

A graph-based functional API for building complex scikit-learn pipelines.

https://baikal.readthedocs.io

BSD 3-Clause "New" or "Revised" License

592 stars 30 forks source link

Closed alegonz closed 4 years ago

alegonz commented 4 years ago

Add capability to specify a name to the class made by make_step.
Fix bug where *args was not included in the constructor of the class made by make_step (and fix related examples in the docs).
Add typing.