TrainingOutputs = typing.NamedTuple(
    "TrainingOutputs",
    model_state=FlyteFile,
)

def train_task(hp: Hyperparameters) -> TrainingOutputs:
    ...
    return TrainingOutputs(model_state=FlyteFile(MODEL_PATH))