name: tutorial

# conda_env: conda.yaml #ścieżka do pliku conda.yaml z definicją środowiska
    
docker_env:
  image: mikolajk/ium:mlflow

entry_points:
  main:
    parameters:
      epochs: {type: float, default: 3}
      learning_rate: {type: float, default: 0.1}
    command: "python ./lab8/trainScript.py {epochs} {learning_rate}"