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"
@@ -18,6 +18,7 @@ requirements = [
'shortuuid>=0.5.0',
'nvidia-ml-py3>=7.352.0',
'python-dateutil>=2.6.1',
+ 'prompt_toolkit==1.0.15',
# Removed until we bring back the board
#'flask-cors>=3.0.3',
#'flask-graphql>=1.4.0',
