Fix requirements.txt max versions

CONTRIBUTING.rst

@@ -80,8 +80,6 @@ Now, here's how to set up `entity-embed` for local development.
     $ pytest
     $ tox
 
-   To get flake8 and tox, just pip install them into your virtualenv.
-
 8. Commit your changes and push your branch to GitHub::
 
     $ git add .

README.md

@@ -34,9 +34,9 @@ https://entity-embed.readthedocs.io
 
 - **Python**: >= 3.6
 - **[Numpy](https://numpy.org/)**: >= 1.19.0
-- **[PyTorch](https://pytorch.org/)**: >= 1.7.1
-- **[PyTorch Lightning](https://pytorch-lightning.readthedocs.io/en/latest/)**: >= 1.1.6
-- **[N2](https://github.com/kakao/n2/)**: >= 0.1.7
+- **[PyTorch](https://pytorch.org/)**: >= 1.7.1, < 1.9
+- **[PyTorch Lightning](https://pytorch-lightning.readthedocs.io/en/latest/)**: >= 1.1.6, < 1.3
+- **[N2](https://github.com/kakao/n2/)**: >= 0.1.7, < 1.2
 
 And others, see [requirements.txt](/requirements.txt).
 

requirements.txt

@@ -1,11 +1,12 @@
-click==7.1.2
-more-itertools>=8.6.0
-n2>=0.1.7
+click==7.1.2,<8.0
+more-itertools>=8.6.0,<9.0
+n2>=0.1.7,<1.2
 numpy>=1.19.0
 ordered-set>=4.0.2
-pytorch_lightning>=1.1.6
-pytorch-metric-learning>=0.9.98
+pytorch_lightning>=1.1.6,<1.3
+pytorch-metric-learning>=0.9.98,<1.0
 regex>=2020.11.13
-torch>=1.7.1
-torchtext>=0.8.1
+torch>=1.7.1,<1.9
+torchtext>=0.8,<0.10
+torchvision>=0.8.2<0.10
 tqdm>=4.53.0