Using Smote in a Pipeline

class_imbalance_pipe.py

from imblearn.pipeline import Pipeline
from imblearn.over_sampling import SMOTE
from sklearn.ensemble import GradientBoostingClassifier

pipeline = Pipeline(steps= [
                    ("SMOTE", SMOTE()),
                    ("GradientBooster", GradientBoostingClassifier())
                    ])

pipeline.fit(X_train, y_train)

y_test_preds = pipeline.predict(X_test)