Describe the solution you'd like
Swap the arguments treatment and y in the fit function.
To change the order of arguments in the fit function from X, treatment, y to X, y, treatment for compatibility with sklearn
Additional context
When using causalml models in the sklearn pipeline, it is necessary to write adapters
Because pipeline expects the following order: X, y, other arguments, and in fit in causalml the following order: X, treatment, y