name: mlflow_464914

conda_env: conda.yaml #ścieżka do pliku conda.yaml z definicją środowisk
# docker_env:
#  image: mlflow-docker-example-environment

entry_points:
  main:
    parameters:
      epochs: {type: int, default: 10}
    command: "python mlflow_model.py {epochs}"
  test:
    command: "python mlflow_prediction.py"