code: def configure_client_for_uc(mock_databricks_uc_host_creds):
      """
      Configure MLflow client to register models to UC
      """
      with mock.patch("databricks_cli.configure.provider.get_config"):
          orig_registry_uri = mlflow.get_registry_uri()
          mlflow.set_registry_uri("databricks-uc")
          yield
          mlflow.set_registry_uri(orig_registry_uri)
  
imports: from unittest import mock
  import pytest
  import mlflow
  from mlflow.utils.rest_utils import MlflowHostCreds
label_desc: Use image node_id 3 for calling a global function with example usage: configure_client_for_uc(mock_databricks_uc_host_creds) without return types