hum, should indexing do the trick here? `self.trial_id = self.trial_id[indices]`