summaryrefslogtreecommitdiff
path: root/examples/pybullet/gym/pybullet_utils/util.py
blob: 5a014c8ed07a8b530a0fc58ca5cf708b435bd654 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
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