From 167767b30c0b28887d996c1a9431d8bd2f93f873 Mon Sep 17 00:00:00 2001
From: Mike Dusenberry <dusenberrymw@google.com>
Date: Mon, 23 Sep 2024 16:23:16 -0700
Subject: [PATCH] Updates a test case for NumPy 2.

PiperOrigin-RevId: 677980827
---
 edward2/tensorflow/random_variable_test.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/edward2/tensorflow/random_variable_test.py b/edward2/tensorflow/random_variable_test.py
index 47da34d6..7359d5b0 100644
--- 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):