27187bfff545ef066da5627d7dff8a6caad19e9e,tensorflow_transform/impl_helper_test.py,ImplHelperTest,testImportAndExportDense,#ImplHelperTest#,168

Before Change

  def testImportAndExportDense(self):
    // Export the function "z = x * y + x + y"
    g = tf.Graph()
    with g.as_default():
      x = tf.placeholder(tf.float32, ())
      y = tf.placeholder(tf.float32, ())
      z = x * y + x + y
    transform_fn_def = impl_helper.make_transform_fn_def(
        g, {"x": x, "y": y}, {"z": z})

    // Import the function, applying it to constants for x and y.
    g = tf.Graph()
    with g.as_default():

After Change

        "y": tf.FixedLenFeature((), tf.float32)
    })

    inputs, outputs = impl_helper.run_preprocessing_fn(
        preprocessing_fn, input_schema)
    saved_model_dir = os.path.join(self.get_temp_dir(), "dense")
    _ = impl_helper.make_transform_fn_def(

Instances