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: == Status ==
  Memory usage on this node: 37.0/480.3 GiB
  Using FIFO scheduling algorithm.
  Resources requested: 0/640 CPUs, 0/8 GPUs, 0.0/2526.95 GiB heap, 0.0/128.42 GiB objects
  Result logdir: /home/ubuntu/ray_results/atari-impala
  Number of trials: 4 (4 TERMINATED)
  +---------------------------------------------+------------+-------+-----------------------------+--------+------------------+-------------+----------+
  | Trial name                                  | status     | loc   | env                         |   iter |   total time (s) |   timesteps |   reward |
  |---------------------------------------------+------------+-------+-----------------------------+--------+------------------+-------------+----------|
  | IMPALA_BreakoutNoFrameskip-v4_511a1830      | TERMINATED |       | BreakoutNoFrameskip-v4      |     39 |      699.024 |     3050000 |   44.21  |
  | IMPALA_BeamRiderNoFrameskip-v4_511b6726     | TERMINATED |       | BeamRiderNoFrameskip-v4     |     37 |      664.972 |     3083500 |  449.72  |
  | IMPALA_QbertNoFrameskip-v4_511bd44a         | TERMINATED |       | QbertNoFrameskip-v4         |     40 |      707.975 |     3088000 |  495.464 |
  | IMPALA_SpaceInvadersNoFrameskip-v4_511c4858 | TERMINATED |       | SpaceInvadersNoFrameskip-v4 |     38 |      680.926 |     3004500 |  380.35  |
  +---------------------------------------------+------------+-------+-----------------------------+--------+------------------+-------------+----------+
  