Make h5py optional

resolve/ubik_tools/plot_sky_hdf5.py

@@ -19,7 +19,6 @@ import os
 import pickle
 from warnings import warn
 
-import h5py
 import matplotlib.pyplot as plt
 import nifty8 as ift
 import numpy as np
@@ -85,6 +84,11 @@ def visualize_sky_hdf5(hdf5_file, output_file, what, stokes, norm="linear", vmin
 
     dpi : int
     """
+    try:
+        import h5py
+    except ImportError:
+        raise ImportError("Need to install the optional dependency h5py")
+
     with h5py.File(hdf5_file, "r") as f:
         if what == "mean":
             if "stats" in f.keys() and "mean" in f["stats"].keys():

setup.py

@@ -142,8 +142,8 @@ setup(
     packages=find_packages(include=["resolve", "resolve.*", "resolve_support", "resolve_support.*"]),
     zip_safe=True,
     dependency_links=[],
-    install_requires=["ducc0", "matplotlib", "h5py", "scipy", "nifty8"],
-    extras_require={"full": ("astropy", "pytest", "pytest-cov", "mpi4py", "python-casacore")},
+    install_requires=["ducc0", "matplotlib", "scipy", "nifty8"],
+    extras_require={"full": ("astropy", "pytest", "pytest-cov", "mpi4py", "python-casacore", "h5py")},
     ext_modules=extensions,
     entry_points={"console_scripts":
         [