Walter_C
Databricks Employee
Databricks Employee

When creating child runs, explicitly set the parent run ID:

def train_model(params):
mlflow.xgboost.autolog()
with mlflow.start_run(nested=True, run_name="child_run", parent_run_id=parent_run.info.run_id):
# Your existing code here