castorini / chatty-goose

A Python framework for conversational search
40 stars 9 forks source link

PyTorch version: needs Torch 1.7 (won't work with 1.8) #13

Closed lintool closed 3 years ago

lintool commented 3 years ago

With a from-scratch installation, the module pulls in Torch 1.8, which causes this error:

ImportError: cannot import name 'SAVE_STATE_WARNING' from 'torch.optim.lr_scheduler' (/anaconda3/envs/chatty-goose-test/lib/python3.7/site-packages/torch/optim/lr_scheduler.py)

Downgrading fixes the issue:

$ pip install torch==1.7.1 torchtext==0.8.1

Should we pin the version in our module dependencies? Or at the very least this needs to be documented.=