[Hotfix] Update TF's random seed API

uwsampl · Nov 25, 2020 · 085752b · 085752b
1 parent 47ef279
commit 085752b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)