def set_num_epochs(self, num_epochs: Union[(int, float)]):
    'Set number of epochs to be run.\n\n        Args:\n           num_epochs: Number of epochs to run task\n        '
    self.num_epochs = num_epochs
    return self
Set number of epochs to be run.

Args:
   num_epochs: Number of epochs to run task
def set_num_epochs(self, num_epochs: Union[(int, float)]):
    'Set number of epochs to be run.\n\n        Args:\n           num_epochs: Number of epochs to run task\n        '
    self.num_epochs = num_epochs
    return self