diff options
Diffstat (limited to 'examples/pybullet/gym/pybullet_utils/util.py')
-rw-r--r-- | examples/pybullet/gym/pybullet_utils/util.py | 19 |
1 files changed, 10 insertions, 9 deletions
diff --git a/examples/pybullet/gym/pybullet_utils/util.py b/examples/pybullet/gym/pybullet_utils/util.py index c11ccb811..5a014c8ed 100644 --- a/examples/pybullet/gym/pybullet_utils/util.py +++ b/examples/pybullet/gym/pybullet_utils/util.py @@ -1,13 +1,14 @@ import random import numpy as np + def set_global_seeds(seed): - try: - import tensorflow as tf - except ImportError: - pass - else: - tf.set_random_seed(seed) - np.random.seed(seed) - random.seed(seed) - return
\ No newline at end of file + try: + import tensorflow as tf + except ImportError: + pass + else: + tf.set_random_seed(seed) + np.random.seed(seed) + random.seed(seed) + return |