hunk.code_window: # In TF >v2, GPU support is included in the base package.
   tensorflow==2.5.0
   tensorflow-probability==0.13.0
   -f https://download.pytorch.org/whl/torch_stable.html
   torch==1.9.0+cu111
   -f https://download.pytorch.org/whl/torch_stable.html
   torchvision==0.10.0+cu111
hunk.labels: keep replace replace keep keep keep keep
hunk.after_edit: tensorflow==2.6.0
   tensorflow-probability==0.14.0
  
hunk.file_path: python/requirements_ml_docker.txt
hunk.type: replace
file: ipython
  
  # Needed for Ray Client error message serialization/deserialization.
  tblib
  
  # In TF >v2, GPU support is included in the base package.
  tensorflow==2.5.0
  tensorflow-probability==0.13.0
  -f https://download.pytorch.org/whl/torch_stable.html
  torch==1.9.0+cu111
  -f https://download.pytorch.org/whl/torch_stable.html
  torchvision==0.10.0+cu111
  