ray/rllib/utils/seed.py

import numpy as np
import random
from ray.rllib.utils import try_import_tf

tf = try_import_tf()


def seed(np_seed=0, random_seed=0, tf_seed=0):
    np.random.seed(np_seed)
    random.seed(random_seed)
    tf.set_random_seed(tf_seed)