Fix custom_transform_and_model
notebook
#651
Merged
custom_transform_and_model
notebook
#651