name: MLflow


docker_env:
  image: szymonjadczak/mlflow:latest

entry_points:
  main:
    parameters:
      epochs: {type: float, default: 3}
    command: "python ./biblioteki_dl.py {epochs}"