name: s444507

docker_env:
  image: adamwojdyla
  volumes: ["/mlflow/tmp/mlruns"]

entry_points:
  main:
    parameters:
      epochs: {type: float, default: 1}
    command: "python3 lab08_deepLearining_mlflow.py {epochs}"