Replace tf.random.set_random_seed() with tf.random.set_seed() from the TF 2.0 API.
@@ -43,7 +43,7 @@ class SerializationUtilsTest(absltest.TestCase):
def test_pack_graph_seed_set_raises(self):
with tf.Graph().as_default() as g:
- tf.random.set_random_seed(1234)
+ tf.compat.v2.random.set_seed(1234)
tf.random.normal([1])
input_value = g.as_graph_def()
with self.assertRaisesRegex(ValueError, 'graph-level random seed'):
