def __repr__(self):
    return f'{self.__class__.__name__}(prefix={self.run_id}, epoch={self.local_epoch})'
def __repr__(self):
    return f'{self.__class__.__name__}(prefix={self.run_id}, epoch={self.local_epoch})'