Thanks, you’re right! Another small error I made here is that method is .numpy()and not .to_numpy. Will update it shortly!