51a0c5fbe8e236cdb7e58afce66d7639d8fe9a45,tensorflow_transform/saved/input_fn_maker_test.py,InputFnMakerTest,_test_build_parsing_transforming_serving_input_fn,#InputFnMakerTest#Any#,269

Before Change

            [outputs["transformed_a"], outputs["transformed_b"]],
            feed_dict=feed_inputs)

        with self.assertRaises(Exception):
          session.run(outputs["transformed_label"])

    self.assertEqual(21, transformed_a[0][0])

After Change

    batch_shape = (len(examples), 1)
    sparse_batch_shape = batch_shape

    if not shape:
      // transformed_b is sparse so _convert_scalars_to_vectors did not fix it
      sparse_batch_shape = sparse_batch_shape[:1]
      transformed_b_dict = dict(zip([tuple(x + [0])
                                     for x in transformed_b.indices.tolist()],
                                    transformed_b.values.tolist()))
    else:
      transformed_b_dict = dict(zip([tuple(x)
                                     for x in transformed_b.indices.tolist()],
                                    transformed_b.values.tolist()))

    self.assertEqual(batch_shape, tuple(transformed_a.shape))
    self.assertEqual(sparse_batch_shape, tuple(transformed_b.dense_shape))

    self.assertEqual(21, transformed_a[0][0])

Instances