Skip to content

Commit

Permalink
Fix dependency issues in tests (icaros-usc#139)
Browse files Browse the repository at this point in the history
* Freeze torch versions in tutorials

* Remove pip install pip command
  • Loading branch information
btjanaka authored Jul 2, 2021
1 parent 091f9d0 commit c1ba194
Show file tree
Hide file tree
Showing 3 changed files with 2 additions and 3 deletions.
2 changes: 1 addition & 1 deletion examples/tutorials/fooling_mnist.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@
"metadata": {},
"outputs": [],
"source": [
"%pip install ribs torch torchvision"
"%pip install ribs torch==1.7 torchvision==0.8"
]
},
{
Expand Down
2 changes: 1 addition & 1 deletion examples/tutorials/lsi_mnist.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@
"metadata": {},
"outputs": [],
"source": [
"%pip install ribs torch torchvision numpy matplotlib"
"%pip install ribs torch==1.7 torchvision==0.8 numpy matplotlib"
]
},
{
Expand Down
1 change: 0 additions & 1 deletion tox.ini
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,6 @@ deps =
coverage: .[all]
benchmarks: .[all]
commands =
pip install -U pip
ribs_core: pytest --basetemp={envtmpdir} tests/archives tests/emitters tests/optimizers tests/factory_test.py
ribs_extras: pytest --basetemp={envtmpdir} tests/visualize_test.py
pin_core: pytest --basetemp={envtmpdir} tests/archives tests/emitters tests/optimizers tests/factory_test.py
Expand Down

0 comments on commit c1ba194

Please sign in to comment.