automatically get latest pytorch version in docs (pyro-ppl#937)

loic001 · Mar 27, 2018 · e9b8e4a · e9b8e4a
1 parent 2a19ace
commit e9b8e4a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -191,4 +191,7 @@ def setup(app):
 
 # @jpchen's hack to get rtd builder to install latest pytorch
 if 'READTHEDOCS' in os.environ:
-    os.system('sh scripts/install_pytorch.sh')
+    os.system('curl -o install.sh https://raw.githubusercontent.com/uber/pyro/dev/scripts/install_pytorch.sh')
+    os.system('curl https://raw.githubusercontent.com/uber/pyro/dev/README.md > README.md')
+    os.system('bash install.sh')
+    os.system('rm -f install.sh')
diff --git a/scripts/install_pytorch.sh b/scripts/install_pytorch.sh
@@ -48,5 +48,6 @@ else
 fi
 
 # Download wheel and install
-wget -P tmp/ "${WHL_PREFIX}/${!WHL_LOOKUP}.whl"
+curl -o tmp/"${!WHL_LOOKUP}.whl" "${WHL_PREFIX}/${!WHL_LOOKUP}.whl"
+
 pip install tmp/*