code: def test_auto_pass_in_exp_config_under_load_run(
      mock_start_job,
      mock_load_tc,
      mock_load_trial,
      mock_load_exp,
      run_obj,
      sagemaker_session,
  ):
      mock_start_job.return_value = _TrainingJob(
          sagemaker_session, "my-job"
      )
      mock_load_tc.return_value = run_obj._trial_component, True
      mock_load_trial.return_value = run_obj._trial
      mock_load_exp.return_value = run_obj._experiment
      with load_run(
          run_name=run_obj.run_name,
          experiment_name=run_obj.experiment_name,
          sagemaker_session=sagemaker_session,
      ):
          estimator = Estimator(
              role="arn:my-role",
              image_uri="my-image",
              sagemaker_session=sagemaker_session,
              output_path=_train_output_path,
          )
          estimator.fit(
              inputs=_train_input_path,
              wait=False,
          )
  
          loaded_run = _RunContext.get_current_run()
          assert loaded_run.run_name == run_obj.run_name
          assert (
              loaded_run.experiment_config == run_obj.experiment_config
          )
  
      expected_exp_config = run_obj.experiment_config
      mock_start_job.assert_called_once_with(
          estimator, _train_input_path, expected_exp_config
      )
  
      # _RunContext is cleaned up after exiting the with statement
      assert not _RunContext.get_current_run()
  
imports: from __future__ import absolute_import
  from unittest.mock import patch, MagicMock
  import pytest
  from sagemaker import Processor
  from sagemaker.estimator import Estimator, _TrainingJob
  from sagemaker.experiments.experiment import Experiment
  from sagemaker.experiments.run import _RunContext
  from sagemaker.experiments import load_run, Run
  from sagemaker.experiments.trial import _Trial
  from sagemaker.experiments.trial_component import _TrialComponent
  from sagemaker.processing import ProcessingJob
  from sagemaker.transformer import _TransformJob, Transformer
  from tests.unit.sagemaker.experiments.helpers import TEST_EXP_NAME, mock_trial_load_or_create_func, mock_tc_load_or_create_func
label_desc: Use image node_id 2 for calling a global function with example usage: test_auto_pass_in_exp_config_under_load_run(mock_start_job, mock_load_tc, mock_load_trial, mock_load_exp, run_obj, sagemaker_session) without return types