Before revision:  The second stage consists of a low-dimensional kernel regression, reducing CATE to a function of the covariate(s) of interest.
After revision:  The second stage consists of a low-dimensional local linear regression, reducing CATE to a function of the covariate(s) of interest.