Updated the documentation and README

KevinMusgrave · Aug 31, 2020 · f133730 · f133730
1 parent f9702cf
commit f133730
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -169,7 +169,7 @@ pip install pytorch-metric-learning --pre
 
 **To install on Windows**:
 ```
-pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html
+pip install torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
 pip install pytorch-metric-learning
 ```
 
@@ -317,7 +317,12 @@ See [powerful-benchmarker](https://github.com/KevinMusgrave/powerful-benchmarker
 
 
 ## Development
-In order to run unit tests do:
+Unit tests can be run with the default ```unittest``` library:
+```bash
+python -m unittest discover
+```
+
+Or if you want to use ```pytest```:
 ```bash
 pip install -e .[dev]
 pytest tests

diff --git a/docs/index.md b/docs/index.md
@@ -117,7 +117,7 @@ pip install pytorch-metric-learning --pre
 
 **To install on Windows**:
 ```
-pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html
+pip install torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
 pip install pytorch-metric-learning
 ```