hunk.code_window: # These requirements are used for the CI and CPU-only Docker images so we install CPU only versions of torch.
   # For GPU Docker images, you should install requirements_ml_docker.txt instead.
   # TODO(amogkam): Remove after https://github.com/tensorflow/tensorflow/issues/52922 is fixed.
   keras==2.6.0
   tensorflow==2.6.0
   tensorflow-probability==0.14.0
  
hunk.labels: keep keep add keep keep keep keep
hunk.after_edit: # If you make changes to this file, please also change `requirements_ml_docker.txt`!
  
hunk.file_path: python/requirements/ml/requirements_dl.txt
hunk.type: add
file: # These requirements are used for the CI and CPU-only Docker images so we install CPU only versions of torch.
  # For GPU Docker images, you should install requirements_ml_docker.txt instead.
  
  # TODO(amogkam): Remove after https://github.com/tensorflow/tensorflow/issues/52922 is fixed.
  keras==2.6.0
  tensorflow==2.6.0
  tensorflow-probability==0.14.0
  
  torch==1.9.0;sys_platform=="darwin"
  torchvision==0.10.0;sys_platform=="darwin"
  
  # On non-OSX machines only install CPU version of torch and torchvision
  -f https://download.pytorch.org/whl/torch_stable.html
  torch==1.9.0+cpu;sys_platform!="darwin"
  -f https://download.pytorch.org/whl/torch_stable.html
  torchvision==0.10.0+cpu;sys_platform!="darwin"