Fix conda build (pytorch#1128)

Summary:
Pull Request resolved: pytorch#1128

Need to add Pyro as a dependency

Reviewed By: saitcakmak

Differential Revision: D35115132

fbshipit-source-id: 6c3e7dcd69422a865528fc4c8bd340c581ad3fcb

.conda/meta.yaml

@@ -17,9 +17,10 @@ requirements:
     - setuptools<48  # setup_requires was deprecated in v48.0.0
     - setuptools-scm
   run:
-    - pytorch>=1.9
-    - gpytorch>=1.6
+    - pytorch >=1.9
+    - gpytorch >=1.6
     - scipy
+    - pyro-ppl 1.8.0
 
 test:
   imports:

.github/workflows/nightly.yml

@@ -96,6 +96,7 @@ jobs:
         conda config --set anaconda_upload no
         conda install -y -c pytorch-nightly pytorch cpuonly
         conda install -y scipy
+        conda install -y -c conda-forge pyro-ppl==1.8.0
         pip install git+https://github.com/cornellius-gp/gpytorch.git
     - name: Build and verify conda package
       shell: bash -l {0}

.github/workflows/test_stable.yml

@@ -48,6 +48,7 @@ jobs:
         conda install -y -c pytorch pytorch cpuonly
         conda install -y pip scipy pytest
         conda install -y -c gpytorch gpytorch
+        conda install -y -c conda-forge pyro-ppl==1.8.0
         pip install .[test]
     - name: Unit tests
       shell: bash -l {0}

environment.yml

@@ -2,7 +2,9 @@ name: botorch
 channels:
   - pytorch
   - gpytorch
+  - conda-forge
 dependencies:
   - pytorch>=1.9
   - gpytorch>=1.6
   - scipy
+  - pyro-ppl=1.8.0

scripts/build_and_verify_conda_package.sh

@@ -15,7 +15,7 @@ export BOTORCH_VERSION
 cur_dir="$(pwd)"
 build_dir="${cur_dir}/conda_build"
 mkdir "${build_dir}"
-conda build -c pytorch -c gpytorch --output-folder "${build_dir}" .
+conda build -c pytorch -c gpytorch -c conda-forge --output-folder "${build_dir}" .
 
 # name of package file (assuming first build)
 path="${build_dir}/noarch/botorch-${BOTORCH_VERSION}-0.tar.bz2"

scripts/test_packaging.sh

@@ -23,5 +23,6 @@ python setup.py sdist bdist_wheel
 conda config --show
 conda config --add channels pytorch
 conda config --add channels gpytorch
+conda config --add channels conda-forge
 cd .conda || exit
 ./build_conda.sh