Commit d63812f1 authored by Asim Shankar's avatar Asim Shankar Committed by TensorFlower Gardener
Browse files

contrib/distributions: Test cleanup. 

- Don't need a test_session() when using
  @test_util.run_in_graph_and_eager_modes
- Can use @test_util.run_all_in_graph_and_eager_modes when all tests in the
  TestCase should be run in both modes.

PiperOrigin-RevId: 209968609
parent 92a7c84e
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment