name: s444465

docker_env:
    image: s444465/ium:mlflow

entry_points:
    main:
        parameters:
            epochs: {type: float, default: 10}
        command: "python mlflow_training.py {epochs}"
    test:
        command: "python mlflow_training.py test"