Fix wandb package broken due to dependency update.
Fix wandb package broken due to dependency update.

We depend on whaaaaat=0.5.2 which depends on prompt_toolkit>=1.0.14

prompt_toolkit just released backwards-incompatible 2.0.1. which
causes an ImportError on this line whenever someone tries to run
the wandb cli: "from prompt_toolkit.token import Token"

wandb/client,wandb/client,wandb/client