Update `pytorch` dep to work with the `py39` tox env (#3686)
Update `pytorch` dep to work with the `py39` tox env (#3686)

* Update `pytorch` dep to work with the `py39` tox env
wandb/client,wandb/client,wandb/client