Skip to content

Commit 409642c

Browse files
committed
Run unittests with and without the faiss library.
faiss is now an optional package for GPyTorch, to be used in conjunction with the VNNGP model (#2026) and hopefully more nearest neighbor GP methods in the future. This PR ensures that our test suite tests the functionality of GPyTorch with and without the faiss library.
1 parent 433b7ee commit 409642c

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

.github/workflows/run_test_suite.yml

+7-2
Original file line numberDiff line numberDiff line change
@@ -37,9 +37,13 @@ jobs:
3737
strategy:
3838
matrix:
3939
pytorch-version: ["master", "stable"]
40-
pyro: ["with-pyro", "no-pyro"]
40+
extras: ["with-extras", "no-extras"]
4141
steps:
4242
- uses: actions/checkout@v2
43+
- uses: conda-incubator/setup-miniconda@v2 # Necessary to install faiss
44+
with:
45+
auto-update-conda: false
46+
python-version: "3.7"
4347
- name: Set up Python
4448
uses: actions/setup-python@v2
4549
with:
@@ -51,8 +55,9 @@ jobs:
5155
else
5256
pip install torch==1.10+cpu -f https://download.pytorch.org/whl/torch_stable.html;
5357
fi
54-
if [[ ${{ matrix.pyro }} == "with-pyro" ]]; then
58+
if [[ ${{ matrix.extras }} == "with-extras" ]]; then
5559
pip install "pyro-ppl>=1.8";
60+
conda install -c pytorch faiss-cpu
5661
fi
5762
pip install -r requirements.txt
5863
- name: Run unit tests

0 commit comments

Comments
 (0)