311a6e809bc4932e3e1c44ea59c557c4931b67d7,tensorflow_transform/beam/cached_impl_test.py,CachedImplTest,test_optimize_traversal,#CachedImplTest#Any#Any#Any#Any#,606

Before Change

                              write_cache_fn, expected_dot_graph_str):
    cache_location = self._make_cache_location()
    span_0_key, span_1_key = "span-0", "span-1"
    if write_cache_fn is not None:
      write_cache_fn(cache_location.input_cache_dir, [span_0_key, span_1_key])

    with tf.name_scope("inputs"):
      input_signature = impl_helper.feature_spec_as_batched_placeholders(
          feature_spec)
    output_signature = preprocessing_fn(input_signature)

After Change

    leaf_nodes = [transform_fn_future] + sorted(
        cache_output_dict.values(), key=str)
    dot_string = nodes.get_dot_graph(leaf_nodes).to_string()
    self.WriteRenderedDotFile(dot_string)

    self.assertSameElements(
        dot_string.split("\n"),

Instances