diff --git a/rltorch/seed.py b/rltorch/seed.py index 74d685b..1b943b7 100644 --- a/rltorch/seed.py +++ b/rltorch/seed.py @@ -4,6 +4,14 @@ import random import torch def set_seed(SEED): + """ + Set the seed for repeatability purposes. + + Parameters + ---------- + SEED : int + The seed to set numpy, random, and torch to. + """ # Set `PYTHONHASHSEED` environment variable at a fixed value environ['PYTHONHASHSEED'] = str(SEED)