What @arjunrajkumar suggests worked for me fine without any errors.
I changed the second df to df_test and kinda duplicated nearly everything afterwards, until we merge back these dataframes into joined and joined_test later in the notebook. 
3 Likes