https://github.com/openai/baselines
Revision 3301089b48c42b87b396e246ea3f56fa4bfc9678 authored by pzhokhov on 26 April 2019, 23:14:49 UTC, committed by GitHub on 26 April 2019, 23:14:49 UTC
* remove bullet extra, constrain gym version to be >= 0.10.0 * constrain gym version from above
1 parent a07fad9
Tip revision: 3301089b48c42b87b396e246ea3f56fa4bfc9678 authored by pzhokhov on 26 April 2019, 23:14:49 UTC
remove bullet extra, constrain gym version to be >= 0.10.0 (#885)
remove bullet extra, constrain gym version to be >= 0.10.0 (#885)
Tip revision: 3301089
setup.py
import re
from setuptools import setup, find_packages
import sys
if sys.version_info.major != 3:
print('This Python is only compatible with Python 3, but you are running '
'Python {}. The installation will likely fail.'.format(sys.version_info.major))
extras = {
'test': [
'filelock',
'pytest',
'pytest-forked',
'atari-py'
],
'mpi': [
'mpi4py'
]
}
all_deps = []
for group_name in extras:
all_deps += extras[group_name]
extras['all'] = all_deps
setup(name='baselines',
packages=[package for package in find_packages()
if package.startswith('baselines')],
install_requires=[
'gym>=0.10.0, <1.0.0',
'scipy',
'tqdm',
'joblib',
'dill',
'progressbar2',
'cloudpickle',
'click',
'opencv-python'
],
extras_require=extras,
description='OpenAI baselines: high quality implementations of reinforcement learning algorithms',
author='OpenAI',
url='https://github.com/openai/baselines',
author_email='gym@openai.com',
version='0.1.5')
# ensure there is some tensorflow build with version above 1.4
import pkg_resources
tf_pkg = None
for tf_pkg_name in ['tensorflow', 'tensorflow-gpu', 'tf-nightly', 'tf-nightly-gpu']:
try:
tf_pkg = pkg_resources.get_distribution(tf_pkg_name)
except pkg_resources.DistributionNotFound:
pass
assert tf_pkg is not None, 'TensorFlow needed, of version above 1.4'
from distutils.version import LooseVersion
assert LooseVersion(re.sub(r'-?rc\d+$', '', tf_pkg.version)) >= LooseVersion('1.4.0')
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...