Unpin pyro. (#2068)

See discussion here: #2059 (comment)

Co-authored-by: Geoff Pleiss <[email protected]>

Author: Balandat

.github/workflows/run_test_suite.yml

@@ -52,7 +52,7 @@ jobs:
           pip install torch==1.10+cpu -f https://download.pytorch.org/whl/torch_stable.html;
         fi
         if [[ ${{ matrix.pyro }} == "with-pyro" ]]; then
-          pip install "pyro-ppl<=1.8";
+          pip install "pyro-ppl>=1.8";
         fi
         pip install -r requirements.txt
     - name: Run unit tests
@@ -72,7 +72,7 @@ jobs:
         pip install pytest nbval jupyter tqdm matplotlib torchvision scipy
         pip install -r requirements.txt
         python setup.py build develop
-        pip install "pyro-ppl<=1.8";
+        pip install "pyro-ppl>=1.8";
     - name: Run example notebooks
       run: |
         grep -l smoke_test examples/**/*.ipynb | xargs grep -L 'smoke_test = False' | CI=true xargs pytest --nbval-lax --current-env

setup.py

@@ -61,7 +61,7 @@ def find_version(*file_paths):
         "dev": ["black", "twine", "pre-commit"],
         "docs": ["ipython", "ipykernel", "sphinx<3.0.0", "sphinx_rtd_theme", "nbsphinx", "m2r"],
         "examples": ["ipython", "jupyter", "matplotlib", "scipy", "torchvision", "tqdm"],
-        "pyro": ["pyro-ppl==1.8"],
+        "pyro": ["pyro-ppl>=1.8"],
         "keops": ["pykeops>=1.1.1"],
         "test": ["flake8==4.0.1", "flake8-print==4.0.0", "pytest", "nbval"],
     },