From 085752bb33a1b04f0ccdc4f3a75b22ab70007336 Mon Sep 17 00:00:00 2001
From: "Steven S. Lyubomirsky" <sslyu@cs.washington.edu>
Date: Tue, 24 Nov 2020 20:22:39 -0800
Subject: [PATCH] [Hotfix] Update TF's random seed API

---
 shared/python/trial_util.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/shared/python/trial_util.py b/shared/python/trial_util.py
index 83efdc4..332e364 100644
--- a/shared/python/trial_util.py
+++ b/shared/python/trial_util.py
@@ -17,7 +17,7 @@ def set_seed(seed):
     # need to have their seeds set
     np.random.seed(seed)
     mx.random.seed(seed)
-    tf.set_random_seed(seed)
+    tf.random.set_seed(seed)
     pt.manual_seed(seed)
     random.seed(seed)