Skip to content

Commit cd774c3

Browse files
committed
Revert "feature: add RL Ray 0.6.5 support (aws#779)"
This reverts commit acbe7cc.
1 parent acbe7cc commit cd774c3

File tree

3 files changed

+17
-23
lines changed

3 files changed

+17
-23
lines changed

src/sagemaker/rl/README.rst

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ With Reinforcement Learning (RL) Estimators, you can train reinforcement learnin
77
Supported versions of Coach: ``0.11.1``, ``0.10.1`` with TensorFlow, ``0.11.0`` with TensorFlow or MXNet.
88
For more information about Coach, see https://github.com/NervanaSystems/coach
99

10-
Supported versions of Ray: ``0.6.5``, ``0.5.3`` with TensorFlow.
10+
Supported versions of Ray: ``0.5.3`` with TensorFlow.
1111
For more information about Ray, see https://github.com/ray-project/ray
1212

1313
For information about using RL with the SageMaker Python SDK, see https://sagemaker.readthedocs.io/en/stable/using_rl.html.
@@ -23,19 +23,19 @@ SageMaker runs RL Estimator scripts in either Python 3.5 for MXNet or Python 3.6
2323

2424
The Docker images have the following dependencies installed:
2525

26-
+-------------------------+-------------------+-------------------+-------------------+-------------------+-------------------+
27-
| Dependencies | Coach 0.10.1 | Coach 0.11.0 | Coach 0.11.1 | Ray 0.5.3 | Ray 0.6.5 |
28-
+-------------------------+-------------------+-------------------+-------------------+-------------------+-------------------+
29-
| Python | 3.6 | 3.5 (MXNet) or | 3.6 | 3.6 | 3.6 |
30-
| | | 3.6 (TensorFlow) | | | |
31-
+-------------------------+-------------------+-------------------+-------------------+-------------------+-------------------+
32-
| CUDA (GPU image only) | 9.0 | 9.0 | 9.0 | 9.0 | 9.0 |
33-
+-------------------------+-------------------+-------------------+-------------------+-------------------+-------------------+
34-
| DL Framework | TensorFlow-1.11.0 | MXNet-1.3.0 or | TensorFlow-1.12.0 | TensorFlow-1.11.0 | TensorFlow-1.12.0 |
35-
| | | TensorFlow-1.11.0 | | | |
36-
+-------------------------+-------------------+-------------------+-------------------+-------------------+-------------------+
37-
| gym | 0.10.5 | 0.10.5 | 0.11.0 | 0.10.5 | 0.12.1 |
38-
+-------------------------+-------------------+-------------------+-------------------+-------------------+-------------------+
26+
+-------------------------+-------------------+-------------------+-------------------+
27+
| Dependencies | Coach 0.10.1 | Coach 0.11.0 | Ray 0.5.3 |
28+
+-------------------------+-------------------+-------------------+-------------------+
29+
| Python | 3.6 | 3.5(MXNet) or | 3.6 |
30+
| | | 3.6(TensorFlow) | |
31+
+-------------------------+-------------------+-------------------+-------------------+
32+
| CUDA (GPU image only) | 9.0 | 9.0 | 9.0 |
33+
+-------------------------+-------------------+-------------------+-------------------+
34+
| DL Framework | TensorFlow-1.11.0 | MXNet-1.3.0 or | TensorFlow-1.11.0 |
35+
| | | TensorFlow-1.11.0 | |
36+
+-------------------------+-------------------+-------------------+-------------------+
37+
| gym | 0.10.5 | 0.10.5 | 0.10.5 |
38+
+-------------------------+-------------------+-------------------+-------------------+
3939

4040
The Docker images extend Ubuntu 16.04.
4141

src/sagemaker/rl/estimator.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -53,13 +53,7 @@
5353
},
5454
'0.5': {
5555
'tensorflow': '1.11'
56-
},
57-
'0.6.5': {
58-
'tensorflow': '1.12'
59-
},
60-
'0.6': {
61-
'tensorflow': '1.12'
62-
},
56+
}
6357
}
6458
}
6559

@@ -79,7 +73,7 @@ class RLEstimator(Framework):
7973

8074
COACH_LATEST_VERSION_TF = '0.11.1'
8175
COACH_LATEST_VERSION_MXNET = '0.11.0'
82-
RAY_LATEST_VERSION = '0.6.5'
76+
RAY_LATEST_VERSION = '0.5.3'
8377

8478
def __init__(self, entry_point, toolkit=None, toolkit_version=None, framework=None,
8579
source_dir=None, hyperparameters=None, image_name=None,

tests/conftest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ def rl_coach_mxnet_version(request):
140140
return request.param
141141

142142

143-
@pytest.fixture(scope='module', params=['0.5', '0.5.3', '0.6', '0.6.5'])
143+
@pytest.fixture(scope='module', params=['0.5', '0.5.3'])
144144
def rl_ray_version(request):
145145
return request.param
146146

0 commit comments

Comments
 (0)