diff --git a/rio_tiler/colormap.py b/rio_tiler/colormap.py index c9e0f768..b0f21239 100644 --- a/rio_tiler/colormap.py +++ b/rio_tiler/colormap.py @@ -16,16 +16,17 @@ ) try: - import importlib.resources as pkg_resources + from importlib.resources import files as resources_files # type: ignore except ImportError: - # Try backported to PY<37 `importlib_resources`. - import importlib_resources as pkg_resources # type: ignore + # Try backported to PY<39 `importlib_resources`. + from importlib_resources import files as resources_files # type: ignore EMPTY_COLORMAP: Dict = {i: [0, 0, 0, 0] for i in range(256)} -with pkg_resources.path(__package__, "cmap_data") as p: - DEFAULT_CMAPS_FILES = {f.stem: str(f) for f in p.glob("*.npy")} +DEFAULT_CMAPS_FILES = { + f.stem: str(f) for f in (resources_files(__package__) / "cmap_data").glob("*.npy") +} USER_CMAPS_DIR = os.environ.get("COLORMAP_DIRECTORY", None) if USER_CMAPS_DIR: diff --git a/setup.py b/setup.py index 3991b551..bcc763d2 100644 --- a/setup.py +++ b/setup.py @@ -17,7 +17,7 @@ "rasterio>=1.1.7", "requests", "rio-color", - "importlib_resources;python_version<'3.7'", + "importlib_resources>=1.1.0;python_version<'3.9'", ] extra_reqs = {