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: // Copyright 2021 The Ray Authors.
  //
  // Licensed under the Apache License, Version 2.0 (the "License");
  // you may not use this file except in compliance with the License.
  // You may obtain a copy of the License at
  //
  //  http://www.apache.org/licenses/LICENSE-2.0
  //
  // Unless required by applicable law or agreed to in writing, software
  // distributed under the License is distributed on an "AS IS" BASIS,
  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  // See the License for the specific language governing permissions and
  // limitations under the License.
  
  class MockStringIdMap : public StringIdMap {};
  