code: def get_huggingface_dataset_builder_mock(load_dataset_builder_mock):
      with mock.patch.object(
          huggingface_dataset_builder, "login_to_hf"
      ) as login_to_hf_mock:
          builder = (
              huggingface_dataset_builder.HuggingfaceDatasetBuilder(
                  file_format="tfrecord",
                  hf_repo_id="foo/bar",
                  hf_config="config",
                  ignore_verifications=True,
                  data_dir="/path/to/data",
                  hf_hub_token="SECRET_TOKEN",
                  hf_num_proc=100,
                  other_arg="this is another arg",
              )
          )
          load_dataset_builder_mock.assert_called_once_with(
              "foo/bar", "config", other_arg="this is another arg"
          )
          login_to_hf_mock.assert_called_once_with("SECRET_TOKEN")
          yield builder
  
imports: import datetime
  from unittest import mock
  from absl import logging
  import numpy
  import pytest
  import tensorflow
  from tensorflow_datasets.core import features
  from tensorflow_datasets.core import lazy_imports_lib
  from tensorflow_datasets.core import registered
  from tensorflow_datasets.core.dataset_builders import huggingface_dataset_builder
label_desc: Use image node_id 15 for calling a global function with example usage: get_huggingface_dataset_builder_mock(load_dataset_builder_mock) without return types