code: def test_fit_kwargs(self, mock_submit_job):
      # TF 1.x is not supported
      if utils.is_tf_v1():
          with self.assertRaises(RuntimeError):
              client.cloud_fit(
                  self._model,
                  x=self._dataset,
                  validation_data=self._dataset,
                  remote_dir=self._remote_dir,
                  job_spec=self._job_spec,
                  batch_size=1,
                  epochs=2,
                  verbose=3,
              )
          return
      job_id = client.cloud_fit(
          self._model,
          x=self._dataset,
          validation_data=self._dataset,
          remote_dir=self._remote_dir,
          region=self._region,
          project_id=self._project_id,
          image_uri=self._image_uri,
          batch_size=1,
          epochs=2,
          verbose=3,
      )
  
      kargs, _ = mock_submit_job.call_args
      body, _ = kargs
      self.assertEqual(body["job_id"], job_id)
      remote_dir = body["trainingInput"]["args"][1]
  
      training_assets_graph = tf.saved_model.load(
          os.path.join(remote_dir, "training_assets")
      )
      elements = training_assets_graph.fit_kwargs_fn()
      self.assertDictContainsSubset(
          tfds.as_numpy(elements),
          {"batch_size": 1, "epochs": 2, "verbose": 3},
      )
  
imports: import os
  import pickle
  import tempfile
  from unittest import mock
  from googleapiclient import discovery
  import tensorflow
  import tensorflow_datasets
  from tensorflow_cloud.tuner import cloud_fit_client
  from tensorflow_cloud.tuner import cloud_fit_utils
  from tensorflow_cloud.utils import google_api_client
label_desc: Use image node_id 7 for calling the CloudFitClientTest obj's underlying member method code with example usage: obj.test_fit_kwargs(mock_submit_job) without return types