[rllib] Fix output API when lz4 not installed (#5421)

ray-project · Aug 10, 2019 · 8b6f0d3 · 8b6f0d3
1 parent de95117
commit 8b6f0d3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/rllib/offline/json_writer.py b/rllib/offline/json_writer.py
@@ -19,7 +19,7 @@
 from ray.rllib.offline.io_context import IOContext
 from ray.rllib.offline.output_writer import OutputWriter
 from ray.rllib.utils.annotations import override, PublicAPI
-from ray.rllib.utils.compression import pack
+from ray.rllib.utils.compression import pack, compression_supported
 
 logger = logging.getLogger(__name__)
 
@@ -99,7 +99,7 @@ def _get_file(self):
 
 
 def _to_jsonable(v, compress):
-    if compress:
+    if compress and compression_supported():
         return str(pack(v))
     elif isinstance(v, np.ndarray):
         return v.tolist()

diff --git a/rllib/utils/compression.py b/rllib/utils/compression.py
@@ -23,6 +23,11 @@
     LZ4_ENABLED = False
 
 
+@DeveloperAPI
+def compression_supported():
+    return LZ4_ENABLED
+
+
 @DeveloperAPI
 def pack(data):
     if LZ4_ENABLED: