def set_model(self, model: ClassyModel):
    'Set model for task\n\n        Args:\n            model: Model to be trained\n        '
    self.base_model = model
    return self
Set model for task

Args:
    model: Model to be trained
def set_model(self, model: ClassyModel):
    'Set model for task\n\n        Args:\n            model: Model to be trained\n        '
    self.base_model = model
    return self