code: def _wait_for(submitted_run_obj):
      """Wait on the passed-in submitted run, reporting its status to the tracking server."""
      run_id = submitted_run_obj.run_id
      active_run = None
      # Note: there's a small chance we fail to report the run's status to the tracking server if
      # we're interrupted before we reach the try block below
      try:
          active_run = (
              tracking.MlflowClient().get_run(run_id)
              if run_id is not None
              else None
          )
          if submitted_run_obj.wait():
              _logger.info("=== Run (ID '%s') succeeded ===", run_id)
              _maybe_set_run_terminated(active_run, "FINISHED")
          else:
              _maybe_set_run_terminated(active_run, "FAILED")
              raise ExecutionException(f"Run (ID '{run_id}') failed")
      except KeyboardInterrupt:
          _logger.error(
              "=== Run (ID '%s') interrupted, cancelling run ===",
              run_id,
          )
          submitted_run_obj.cancel()
          _maybe_set_run_terminated(active_run, "FAILED")
          raise
  
imports: import json
  import logging
  import os
  import yaml
  import mlflow.projects.databricks
  import mlflow.utils.uri
  from mlflow import tracking
  from mlflow.entities import RunStatus
  from mlflow.exceptions import ExecutionException, MlflowException
  from mlflow.projects.backend import loader
  from mlflow.projects.submitted_run import SubmittedRun
  from mlflow.projects.utils import MLFLOW_LOCAL_BACKEND_RUN_ID_CONFIG, PROJECT_BUILD_IMAGE, PROJECT_DOCKER_ARGS, PROJECT_DOCKER_AUTH, PROJECT_ENV_MANAGER, PROJECT_STORAGE_DIR, PROJECT_SYNCHRONOUS, fetch_and_validate_project, get_entry_point_command, get_or_create_run, get_run_env_vars, load_project
  from mlflow.tracking.fluent import _get_experiment_id
  from mlflow.utils import env_manager
  from mlflow.utils.mlflow_tags import MLFLOW_DOCKER_IMAGE_ID, MLFLOW_PROJECT_BACKEND, MLFLOW_PROJECT_ENV, MLFLOW_RUN_NAME
label_desc: Use image node_id 4 for calling a global function with example usage: _wait_for(submitted_run_obj) without return types