def METHOD_NAME(self, shared_state):
    aiplatform.init(
        project=e2e_base._PROJECT,
        location=e2e_base._LOCATION,
        staging_bucket=shared_state["staging_bucket_name"],
        experiment=self._experiment_name,
        experiment_tensorboard=self._backing_tensorboard,
    )
    shared_state["resources"].append(self._backing_tensorboard)
    shared_state["resources"].append(
        aiplatform.metadata.metadata._experiment_tracker.experiment
    )
    display_name = self._make_display_name("custom-job")
    custom_job = aiplatform.CustomJob.from_local_script(
        display_name=display_name,
        script_path=_LOCAL_TRAINING_SCRIPT_PATH,
        container_uri=_PREBUILT_CONTAINER_IMAGE,
        requirements=["scikit-learn", "pandas"],
        enable_autolog=True,
    )
    custom_job.run(
        experiment=self._experiment_name,
        experiment_run=self._experiment_run_name,
        service_account=self._service_account,
    )
    shared_state["resources"].append(custom_job)
    assert custom_job.state == gca_job_state.JobState.JOB_STATE_SUCCEEDED