Conor McGregor caught joking around with training partner during gym session