return_components for R-Learner

[baendigreydo]

I wanted to receive to components of the CATE prediction via R-Learner. Unfortunately, it is not possible to retrieve this information because it is not defined here. I tried to solve it on my own but I was not successful because it is unclear which "predict"-function is used in Line 134 - dhat = self.models_tau[group].predict(X). Which predict is used for calculation?

It would be very useful to conduct a validation of all Meta Learners in the here proposed way.