torch
pyyaml
wandb
urllib3==1.22