Just implement fit_transform and remove transform.