code: def testGermanCreditNumericInXLA(self):
      @tf.function(autograph=False, jit_compile=True)
      def load_dataset_in_xla():
          dataset = data.german_credit_numeric()
          # The actual dataset loading will happen in Eager mode, courtesy of the
          # init_scope.
          return (
              tf.convert_to_tensor(dataset["train_features"]),
              tf.convert_to_tensor(dataset["train_labels"]),
          )
  
      load_dataset_in_xla()
  
file_path: probability/spinoffs/inference_gym/inference_gym/internal/data_test.py
imports: import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import test_util
  from inference_gym.internal import data
  from inference_gym.internal import test_util
label_desc: Use image node_id 2 for calling the DataTest obj's underlying member method code with example usage: obj.testGermanCreditNumericInXLA() and returns: tf,  tf