fix build

65c0b326 · Yuxin Wu · f63e0ee4 · 65c0b326
Commit 65c0b326 authored Sep 01, 2018 by Yuxin Wu
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

tensorpack/utils/serialize.py tensorpack/utils/serialize.py +2 -0

No files found.
--- a/tensorpack/utils/serialize.py
+++ b/tensorpack/utils/serialize.py
@@ -49,6 +49,8 @@ try:
    # So we need an option to disable it.
    if os.environ.get('TENSORPACK_SERIALIZE', 'pyarrow') == 'pyarrow':
        import pyarrow as pa
+    else:
+        pa = None
 except ImportError:
    pa = None
    dumps_pyarrow = create_dummy_func('dumps_pyarrow', ['pyarrow'])  # noqa