code: def test_predict_uses_registry_uri(
      tmp_recipe_root_path: Path,
      predict_step_output_dir: Path,
      registry_uri_path: Path,
  ):
      registry_uri = registry_uri_path
      model_name = "model_" + get_random_id()
      mlflow.set_registry_uri(registry_uri)
      model_uri = train_log_and_register_model(
          model_name, is_dummy=True
      )
      # reset model registry
      mlflow.set_registry_uri("")
  
      recipe_config = read_yaml(
          tmp_recipe_root_path, _RECIPE_CONFIG_FILE_NAME
      )
      recipe_config.update(
          {"model_registry": {"registry_uri": str(registry_uri)}}
      )
      recipe_config.update(
          {
              "steps": {
                  "predict": {
                      "output": {
                          "using": "parquet",
                          "location": str(
                              predict_step_output_dir.joinpath(
                                  "output.parquet"
                              )
                          ),
                      },
                      "model_uri": model_uri,
                  },
              },
          }
      )
      PredictStep.from_recipe_config(
          recipe_config,
          str(tmp_recipe_root_path),
      ).run(str(predict_step_output_dir))
      assert mlflow.get_registry_uri() == registry_uri
      prediction_assertions(
          predict_step_output_dir, "parquet", "output", spark_session
      )
  
imports: import os
  import tempfile
  from pathlib import Path
  from unittest import mock
  import pandas
  import pytest
  from pyspark.sql import SparkSession
  from sklearn.datasets import load_diabetes
  import mlflow
  from mlflow.exceptions import MlflowException
  from mlflow.recipes.artifacts import RegisteredModelVersionInfo
  from mlflow.recipes.steps.predict import _INPUT_FILE_NAME, _SCORED_OUTPUT_FILE_NAME, PredictStep
  from mlflow.recipes.steps.register import _REGISTERED_MV_INFO_FILE
  from mlflow.recipes.utils import _RECIPE_CONFIG_FILE_NAME
  from mlflow.utils.file_utils import read_yaml
  from tests.recipes.helper_functions import get_random_id, train_and_log_model, train_log_and_register_model
label_desc: Use image node_id 14 for calling a global function with example usage: test_predict_uses_registry_uri(tmp_recipe_root_path, predict_step_output_dir, registry_uri_path) without return types