Updates a test case for NumPy 2.

PiperOrigin-RevId: 677980827
google · Sep 23, 2024 · 167767b · 167767b
1 parent 1600fa7
commit 167767b
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/edward2/tensorflow/random_variable_test.py b/edward2/tensorflow/random_variable_test.py
@@ -107,8 +107,12 @@ def testStr(self):
 
   def testRepr(self):
     x = ed.RandomVariable(tfp.distributions.Normal(0.0, 1.0), value=1.234)
-    string = ("<ed.RandomVariable '{name}' shape=() "
-              "dtype=float32 numpy=1.234>".format(name=x.distribution.name))
+    string = (
+        "<ed.RandomVariable '{name}' shape=() "
+        "dtype=float32 numpy=np.float32(1.234)>".format(
+            name=x.distribution.name
+        )
+    )
     self.assertEqual(repr(x), string)
 
   def testNumpy(self):