Baikal uses an adapter pattern where it dynamically extends the functionality of classes. However, it's make_step method fails because sklearn_pandas has setstate_ method but not getstate method. Also from consistency perspective it make sense to have both these methods implemented. One option is to get rid of setstate method but that will break backward compatibility.
@dukebody -- I would love to get your perspective on this.