Have TensorFlow Distributions share name scopes across method calls.

PiperOrigin-RevId: 193428925
aymar73 · Apr 18, 2018 · fd122fa · fd122fa
1 parent 34230a6
commit fd122fa
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tensorflow_probability/python/edward2/random_variable_test.py b/tensorflow_probability/python/edward2/random_variable_test.py
@@ -85,18 +85,18 @@ def testStr(self):
     if tfe.in_eager_mode():
       pattern = "RandomVariable(\"1.234\", shape=(), dtype=float32"
     else:
-      pattern = "RandomVariable(\"Normal\", shape=(), dtype=float32"
+      pattern = "RandomVariable(\"Normal/\", shape=(), dtype=float32"
     regexp = re.escape(pattern)
     self.assertRegexpMatches(str(x), regexp)
 
   @tfe.run_test_in_graph_and_eager_modes()
   def testRepr(self):
     x = ed.RandomVariable(tfd.Normal(0.0, 1.0), value=1.234)
     if tfe.in_eager_mode():
-      string = ("<ed.RandomVariable 'Normal' shape=() dtype=float32 "
-                "numpy=1.234>")
+      string = ("<ed.RandomVariable 'Normal/' shape=() "
+                "dtype=float32 numpy=1.234>")
     else:
-      string = "<ed.RandomVariable 'Normal' shape=() dtype=float32>"
+      string = "<ed.RandomVariable 'Normal/' shape=() dtype=float32>"
     self.assertEqual(repr(x), string)
 
   @tfe.run_test_in_graph_and_eager_modes()