code: def dataset_source_registry():
      registry = DatasetSourceRegistry()
      with mock.patch(
          "mlflow.data.dataset_source_registry._dataset_source_registry",
          wraps=registry,
      ):
          yield registry
  
file_path: mlflow/tests/data/test_dataset_registry.py
imports: from typing import Optional
  from unittest import mock
  import pytest
  import mlflow.data
  from mlflow.data.dataset import Dataset
  from mlflow.data.dataset_registry import DatasetRegistry, register_constructor
  from mlflow.data.dataset_source_registry import DatasetSourceRegistry, resolve_dataset_source
  from mlflow.exceptions import MlflowException
  from tests.resources.data.dataset import TestDataset
  from tests.resources.data.dataset_source import TestDatasetSource
label_desc: Use image node_id 1 for calling a global function with example usage: dataset_source_registry() without return types