Catch missing pyarrow import

See dask/fastparquet#142
mrocklin · May 4, 2017 · b7be498 · b7be498
1 parent 9ef62f2
commit b7be498
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/dask/dataframe/io/parquet.py b/dask/dataframe/io/parquet.py
@@ -438,8 +438,12 @@ def normalize_ParquetFile(pf):
         return (type(pf), pf.fn, pf.sep) + normalize_token(pf.open)
 
 
-if pyarrow_parquet:
-    @partial(normalize_token.register, pyarrow_parquet.ParquetDataset)
+try:
+    from pyarrow.parquet import ParquetDataset
+except ImportError:
+    pass
+else:
+    @partial(normalize_token.register, ParquetDataset)
     def normalize_PyArrowParquetDataset(ds):
         return (type(ds), ds.paths)