wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-16 11:21:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

[rllib] Remove dependency on TensorFlow (#4764)

Browse Source 
* remove hard tf dep

* add test

* comment fix

* fix test
This commit is contained in:
Eric Liang
2019-05-10 20:36:18 -07:00
committed by GitHub
parent ccc540adf1
commit 351753aae5
35 changed files with 189 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/logger.py
+8 -4
View File
@@ -118,10 +118,14 @@ class TFLogger(Logger):
def _init(self):
try:
global tf, use_tf150_api
import tensorflow
tf = tensorflow
use_tf150_api = (distutils.version.LooseVersion(tf.VERSION) >=
distutils.version.LooseVersion("1.5.0"))
if "RLLIB_TEST_NO_TF_IMPORT" in os.environ:
logger.warning("Not importing TensorFlow for test purposes")
tf = None
else:
import tensorflow
tf = tensorflow
use_tf150_api = (distutils.version.LooseVersion(tf.VERSION) >=
distutils.version.LooseVersion("1.5.0"))
except ImportError:
logger.warning("Couldn't import TensorFlow - "
"disabling TensorBoard logging.")