38e9391dcb5e8991a1a663601ce78bf7f08df474,tensorflow_transform/graph_tools_test.py,GraphToolsTest,testGetDependentInputs,#GraphToolsTest#Any#Any#Any#Any#,551

Before Change

  )
  def testGetDependentInputs(self, create_graph_fn, feeds, fetches,
                             expected_dependent_inputs):
    tensors = create_graph_fn()
    got = graph_tools.get_dependent_inputs(tf.compat.v1.get_default_graph(),
                                           {x: tensors[x] for x in feeds},
                                           {y: tensors[y] for y in fetches})
    self.assertCountEqual(expected_dependent_inputs, got.keys())
    for input_name in expected_dependent_inputs:
      self.assertEqual(tensors[input_name], got[input_name])

After Change

  )
  def testGetDependentInputs(self, create_graph_fn, feeds, fetches,
                             expected_dependent_inputs):
    with tf.compat.v1.Graph().as_default() as graph:
      tensors = create_graph_fn()
      got = graph_tools.get_dependent_inputs(graph,
                                             {x: tensors[x] for x in feeds},
                                             {y: tensors[y] for y in fetches})
    self.assertCountEqual(expected_dependent_inputs, got.keys())
    for input_name in expected_dependent_inputs:
      self.assertEqual(tensors[input_name], got[input_name])

Instances