earlier doc reference in readme

mfinzi · Mar 20, 2021 · cdcbd22 · cdcbd22
1 parent c774c40
commit cdcbd22
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -8,7 +8,7 @@
 [![Documentation](https://readthedocs.org/projects/emlp/badge/)](https://emlp.readthedocs.io/en/latest/) | **[Paper](https://www.youtube.com/watch?v=dQw4w9WgXcQ&ab_channel=RickAstleyVEVO)**| [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/mfinzi/equivariant-MLP/blob/master/docs/notebooks/colabs/all.ipynb)
 
 
-*EMLP* is a jax library for the automated construction of equivariant layers in deep learning.
+*EMLP* is a jax library for the automated construction of equivariant layers in deep learning. You can read the documentation [here](https://emlp.readthedocs.io/en/latest/).
 
 *WARNING*: Our library (and paper) have not yet been released, and may have sharp edges, bugs, and may be subject to breaking changes. 
 Use at your own caution. But if you notice things behaving unexpectedly or get frustrated, send me an email so I can make the library better.

diff --git a/emlp/models/mlp.py b/emlp/models/mlp.py
@@ -172,7 +172,7 @@ class EMLP(Module,metaclass=Named):
             rep_in (Rep): input representation
             rep_out (Rep): output representation
             group (Group): symmetry group
-            ch (int or List[int] or Rep or List[Rep]): number of channels in the hidden layers
+            ch (int or list[int] or Rep or list[Rep]): number of channels in the hidden layers
             num_layers (int): number of hidden layers
 
         Returns: