Skip to content

Commit 57beda9

Browse files
committed
install smexperiments when python >= 3.6
1 parent d87e930 commit 57beda9

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

setup.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -54,13 +54,13 @@ def read_version():
5454
],
5555

5656
install_requires=['sagemaker-containers>=2.6.2', 'numpy', 'scipy', 'sklearn',
57-
'pandas', 'Pillow', 'h5py'],
57+
'pandas', 'Pillow', 'h5py',
58+
'sagemaker-experiments==0.1.7;python_version>="3.6"'],
5859
extras_require={
5960
'test': ['tox', 'flake8', 'pytest', 'pytest-cov', 'pytest-xdist', 'mock',
6061
'sagemaker==1.50.1', 'tensorflow<2.0', 'docker-compose', 'boto3==1.10.50',
6162
'six==1.13.0', 'python-dateutil>=2.1,<2.8.1', 'botocore==1.13.50',
6263
'requests-mock', 'awscli==1.16.314'],
63-
'benchmark': ['click'],
64-
':python_version=="3.6"': ['sagemaker-experiments==0.1.7']
64+
'benchmark': ['click']
6565
},
6666
)

0 commit comments

Comments
 (0)