job.py

# description: train fastai resnet18 model on mnist data via mlflow mlproject

# imports
import mlflow

from pathlib import Path
from azureml.core import Workspace

# get workspace
ws = Workspace.from_config()

# get root of git repo
prefix = Path(__file__).parent

# project settings
project_uri = prefix.joinpath("src")

# azure ml settings
experiment_name = "fastai-mnist-mlproject-example"
compute_name = "cpu-cluster"

# setup mlflow tracking
mlflow.set_tracking_uri(ws.get_mlflow_tracking_uri())
mlflow.set_experiment(experiment_name)

# setup backend config
backend_config = {"COMPUTE": compute_name}

# run mlflow project
run = mlflow.projects.run(
    uri=str(project_uri), backend="azureml", backend_config=backend_config
)