code: def test_in_memory_data(self):
      # This test should only run in tf 2.x
      if utils.is_tf_v1():
          return
  
      # Create a folder under remote dir for this test's data
      tmp_folder = str(uuid.uuid4())
      remote_dir = os.path.join(self._remote_dir, tmp_folder)
  
      # Keep track of test folders created for final clean up
      self._test_folders.append(remote_dir)
  
      x = np.random.random((2, 3))
      y = np.random.randint(0, 2, (2, 2))
  
      job_id = client.cloud_fit(
          self._model(),
          x=x,
          y=y,
          remote_dir=remote_dir,
          region=self._region,
          project_id=self._project_id,
          image_uri=self._image_uri,
          job_id="cloud_fit_e2e_test_{}_{}".format(
              _BUILD_ID.replace("-", "_"), "test_in_memory_data"
          ),
          epochs=2,
      )
      logging.info(
          "test_in_memory_data submitted with job id: %s", job_id
      )
  
      # Wait for AIP Training job to finish successfully
      self.assertTrue(
          google_api_client.wait_for_aip_training_job_completion(
              job_id, self._project_id
          )
      )
  
      # load model from remote dir
      trained_model = tf.keras.models.load_model(
          os.path.join(remote_dir, "checkpoint")
      )
      eval_results = trained_model.evaluate(x, y)
  
      # Accuracy should be better than zero
      self.assertListEqual(
          trained_model.metrics_names, ["loss", "accuracy"]
      )
      self.assertGreater(eval_results[1], 0)
  
file_path: cloud/src/python/tensorflow_cloud/tuner/tests/integration/cloud_fit_integration_test.py
imports: import os
  from typing import Text
  import uuid
  from absl import logging
  import numpy
  import tensorflow
  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 6 for calling the CloudFitIntegrationTest obj's underlying member method code with example usage: obj.test_in_memory_data() without return types