27187bfff545ef066da5627d7dff8a6caad19e9e,tensorflow_transform/impl_helper_test.py,ImplHelperTest,testImportAndExportSparse,#ImplHelperTest#,206

Before Change

  def testImportAndExportSparse(self):
    // Export the function "z = x + y"
    g = tf.Graph()
    with g.as_default():
      x = tf.sparse_placeholder(tf.float32)
      y = tf.sparse_placeholder(tf.float32)
      z = tf.sparse_add(x, y)

After Change

      return {
          "z": api.map(tf.sparse_add, inputs["x"], inputs["y"])
      }
    input_schema = self.toSchema({
        "x": tf.VarLenFeature(tf.float32),
        "y": tf.VarLenFeature(tf.float32)
    })

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

Instances