fix: typehint with | and BaseEstimator from sklearn to avoid error du… …ring build of the documentation