Move datacube imports to function level

Author: robbibt

intertidal/composites.py

@@ -2,11 +2,9 @@
 import sys
 
 import click
-import datacube
 import numpy as np
 import odc.geo.xr
 import xarray as xr
-from datacube.utils.aws import configure_s3_access
 from dea_tools.dask import create_local_dask_cluster
 from eo_tides.eo import pixel_tides
 from odc.algo import (
@@ -439,6 +437,18 @@ def tidal_composites_cli(
     aws_unsigned,
     overwrite,
 ):
+    # Attempt to import datacube and raise an error if not available
+    try:
+        import datacube
+        from datacube.utils.aws import configure_s3_access
+    except ImportError as e:
+        msg = (
+            "The DEA Tidal Composites CLI is configured for Australian applications, and "
+            "requires `datacube`. Please install DEA Intertidal with the "
+            "`[datacube]` extra, e.g.: `pip install dea-intertidal[datacube]`"
+        )
+        raise ImportError(msg) from e
+
     # Create sample filename to test if data exists on file system
     filename = f"{output_dir}ga_s2_tidal_composites_cyear_3/{output_version.replace('.', '-')}/{study_area[:4]}/{study_area[4:]}/{label_date}--P1Y/ga_s2_tidal_composites_cyear_3_{study_area}_{label_date}--P1Y_final.stac-item.json"
 

intertidal/elevation.py

@@ -3,12 +3,10 @@
 from itertools import repeat
 
 import click
-import datacube
 import matplotlib.pyplot as plt
 import numpy as np
 import seaborn as sns
 import xarray as xr
-from datacube.utils.aws import configure_s3_access
 from dea_tools.dask import create_local_dask_cluster
 from eo_tides.eo import pixel_tides
 from odc.algo import xr_quantile
@@ -1125,6 +1123,18 @@ def intertidal_cli(
     exposure_offsets,
     aws_unsigned,
 ):
+    # Attempt to import datacube and raise an error if not available
+    try:
+        import datacube
+        from datacube.utils.aws import configure_s3_access
+    except ImportError as e:
+        msg = (
+            "The DEA Intertidal CLI is configured for Australian applications, and "
+            "requires `datacube`. Please install DEA Intertidal with the "
+            "`[datacube]` extra, e.g.: `pip install dea-intertidal[datacube]`"
+        )
+        raise ImportError(msg) from e
+
     # Create a unique run ID for analysis based on input params and use
     # for logs
     input_params = locals()

intertidal/exposure.py

@@ -4,7 +4,6 @@
 import numpy as np
 import pandas as pd
 import pytz
-import sunriset
 import xarray as xr
 from eo_tides.eo import _pixel_tides_resample, pixel_tides
 
@@ -38,6 +37,9 @@ def temporal_filters(x, time_range, dem):
         timesteps.
 
     """
+    # Import sunriset here to avoid dependency issues if function not used
+    import sunriset
+
     if x == "dry":
         return time_range.drop(
             time_range[

intertidal/io.py

@@ -11,8 +11,6 @@
 import numpy as np
 import odc.geo.xr
 import xarray as xr
-from datacube.utils.geometry import Geometry as Geometry_datacube18
-from datacube.utils.masking import mask_invalid_data
 from dea_tools.coastal import glint_angle
 from eo_tides.stats import tide_stats
 from eodatasets3 import DatasetAssembler, serialise
@@ -116,8 +114,13 @@ def extract_geobox(
         the CRS, resolution, shape and extent of the study area).
 
     """
-    # List of valid input geometry types (from `odc-geo` or `datacube-core`)
-    GEOM_TYPES = (odc.geo.geom.Geometry, Geometry_datacube18)
+    # List of valid input geometry types (from `odc-geo` or `datacube`).
+    # If `datacube` is not installed, only support `odc-geo` geometries
+    try:
+        from datacube.utils.geometry import Geometry as Geometry_datacube18
+        GEOM_TYPES = (odc.geo.geom.Geometry, Geometry_datacube18)
+    except ImportError:
+        GEOM_TYPES = (odc.geo.geom.Geometry)
 
     # Either `study_area` or `geom` must be provided
     if study_area is None and geom is None:
@@ -253,6 +256,19 @@ def load_data(
         to generate ODC lineage metadata for DEA Intertidal)
 
     """
+
+    # Attempt to import datacube and raise an error if not available
+    try:
+        from datacube.utils.masking import mask_invalid_data
+    except ImportError as e:
+        msg = (
+            "The `load_data` function requires `datacube` to be installed. "
+            "Please consider loading data with `odc-stac` instead, or install "
+            "DEA Intertidal with the `[datacube]` extra, e.g.: `pip install "
+            "dea-intertidal[datacube]`"
+        )
+        raise ImportError(msg) from e
+
     # Set spectral bands to load
     s2_spectral_bands = [
         "nbart_blue",