minor fixes from internal (#665)
* sync internal changes. Make ddpg work with vecenvs * B -> nenvs for consistency with other algos, small cleanups * eval_done[d]==True -> eval_done[d] * flake8 and numpy.random.random_integers deprecation warning * Merge branch 'master' of github.com:openai/games into peterz_track_baselines_branch
This commit is contained in:
16
setup.py
16
setup.py
@@ -48,9 +48,13 @@ setup(name='baselines',
|
||||
|
||||
|
||||
# ensure there is some tensorflow build with version above 1.4
|
||||
try:
|
||||
from distutils.version import StrictVersion
|
||||
import tensorflow
|
||||
assert StrictVersion(re.sub(r'-rc\d+$', '', tensorflow.__version__)) >= StrictVersion('1.4.0')
|
||||
except ImportError:
|
||||
assert False, "TensorFlow needed, of version above 1.4"
|
||||
import pkg_resources
|
||||
tf_pkg = None
|
||||
for tf_pkg_name in ['tensorflow', 'tensorflow-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 StrictVersion
|
||||
assert StrictVersion(re.sub(r'-rc\d+$', '', tf_pkg.version)) >= StrictVersion('1.4.0')
|
||||
|
Reference in New Issue
Block a user