@Adrian

I totally resonate with your thought here. Data leakage could have been avoided here by splitting the data earlier in the stage.

I will make the changes.