code: def test_log_model_calls_register_model(tmp_path, jsl_model):
      artifact_path = "model"
      dfs_tmp_dir = tmp_path.joinpath("test")
      register_model_patch = mock.patch(
          "mlflow.tracking._model_registry.fluent._register_model"
      )
      with mlflow.start_run(), register_model_patch:
          mlflow.johnsnowlabs.log_model(
              artifact_path=artifact_path,
              spark_model=jsl_model,
              dfs_tmpdir=dfs_tmp_dir,
              registered_model_name="AdsModel1",
          )
          model_uri = (
              f"runs:/{mlflow.active_run().info.run_id}/{artifact_path}"
          )
          assert_register_model_called_with_local_model_path(
              register_model_mock=mlflow.tracking._model_registry.fluent._register_model,
              model_uri=model_uri,
              registered_model_name="AdsModel1",
          )
  
imports: import json
  import os
  from pathlib import Path
  from unittest import mock
  import pandas
  import pyspark
  import pytest
  from johnsnowlabs import nlp
  from packaging.version import Version
  import mlflow
  import mlflow.utils.file_utils
  from mlflow import pyfunc
  from mlflow.environment_variables import MLFLOW_DFS_TMP
  from mlflow.models import Model, infer_signature
  from mlflow.models.utils import _read_example
  from mlflow.pyfunc import spark_udf
  from mlflow.utils.environment import _mlflow_conda_env
  from mlflow.utils.file_utils import TempDir
  from tests.helper_functions import assert_register_model_called_with_local_model_path
label_desc: Use image node_id 13 for calling a global function with example usage: test_log_model_calls_register_model(tmp_path, jsl_model) without return types