Move _zarr_async to utils module

Author: charles-turner-1

intake_esm/source.py

@@ -6,7 +6,6 @@
 import pandas as pd
 import pydantic
 import xarray as xr
-import zarr
 from intake.source.base import DataSource, Schema
 
 from .cat import Aggregation, DataFormat
@@ -21,15 +20,6 @@ class ESMDataSourceError(Exception):
     pass
 
 
-def _zarr_async() -> bool:
-    """
-    Zarr went all async in version 3.0.0. This sets the async flag based on
-    the zarr version in storage options
-    """
-
-    return int(zarr.__version__.split('.')[0]) > 2
-
-
 def _get_xarray_open_kwargs(data_format, xarray_open_kwargs=None, storage_options=None):
     xarray_open_kwargs = (xarray_open_kwargs or {}).copy()
     _default_open_kwargs = {

intake_esm/utils.py

@@ -5,6 +5,7 @@
 from collections import defaultdict
 
 import polars as pl
+import zarr
 
 
 def show_versions(file=sys.stdout):  # pragma: no cover
@@ -57,6 +58,15 @@ def show_versions(file=sys.stdout):  # pragma: no cover
         print(f'{k}: {stat}', file=file)
 
 
+def _zarr_async() -> bool:
+    """
+    Zarr went all async in version 3.0.0. This sets the async flag based on
+    the zarr version in storage options
+    """
+
+    return int(zarr.__version__.split('.')[0]) > 2
+
+
 OPTIONS = {
     'attrs_prefix': 'intake_esm_attrs',
     'dataset_key': 'intake_esm_dataset_key',

tests/test_source.py

@@ -14,8 +14,8 @@
     _get_xarray_open_kwargs,
     _open_dataset,
     _update_attrs,
-    _zarr_async,
 )
+from intake_esm.utils import _zarr_async
 
 dask.config.set(scheduler='single-threaded')