[DOC] Small improvements

docs/source/conf.py

@@ -180,6 +180,9 @@ texinfo_documents = [
 intersphinx_mapping = {
     "python": ("https://docs.python.org/", None),
     "numpy": ("https://docs.scipy.org/doc/numpy/", None),
+    "torch": ('https://pytorch.org/docs/stable/', None),
+    "pytorch_lightning":
+    ("https://pytorch-lightning.readthedocs.io/en/stable/", None),
 }
 
 # -- Options for Epub output ----------------------------------------------

prototorch/datasets/spiral.py

@@ -5,6 +5,10 @@ import torch
 
 
 def make_spiral(n_samples=500, noise=0.3):
+    """Generates the Spiral Dataset.
+    
+    For use in Prototorch use `prototorch.datasets.Spiral` instead.
+    """
     def get_samples(n, delta_t):
         points = []
         for i in range(n):