Import rioxarray to register rio accessor

Xref #3323
GenericMappingTools · Sep 30, 2024 · 66183ed · 66183ed
1 parent f11decc
commit 66183ed
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/pygmt/datasets/earth_day.py b/pygmt/datasets/earth_day.py
@@ -5,12 +5,17 @@
 The images are available in various resolutions.
 """
 
+import contextlib
 from collections.abc import Sequence
 from typing import Literal
 
 import xarray as xr
 from pygmt.datasets.load_remote_dataset import _load_remote_dataset
 
+with contextlib.suppress(ImportError):
+    # rioxarray is needed to register the rio accessor
+    import rioxarray  # noqa: F401
+
 __doctest_skip__ = ["load_blue_marble"]
 
 

diff --git a/pygmt/datasets/earth_night.py b/pygmt/datasets/earth_night.py
@@ -5,12 +5,17 @@
 The images are available in various resolutions.
 """
 
+import contextlib
 from collections.abc import Sequence
 from typing import Literal
 
 import xarray as xr
 from pygmt.datasets.load_remote_dataset import _load_remote_dataset
 
+with contextlib.suppress(ImportError):
+    # rioxarray is needed to register the rio accessor
+    import rioxarray  # noqa: F401
+
 __doctest_skip__ = ["load_black_marble"]