Custom Transformer in Pipeline with GridSearchCV | AlgoML Prep