From a47f3f603768f2c7de178e583bb44ed311b138f6 Mon Sep 17 00:00:00 2001
From: Brandon Rozek <rozekbrandon@gmail.com>
Date: Fri, 20 Mar 2020 19:38:35 -0400
Subject: [PATCH] Seed documentation

---
 rltorch/seed.py | 8 ++++++++
 1 file changed, 8 insertions(+)

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)