|
1 | 1 | import functools |
2 | | -from typing import List, Literal, Mapping, Union |
| 2 | +from collections.abc import Mapping |
| 3 | +from typing import Literal |
3 | 4 |
|
4 | 5 | import pystac |
5 | 6 | import xarray |
|
10 | 11 | @functools.singledispatch |
11 | 12 | def to_xarray( |
12 | 13 | obj, |
13 | | - stacking_library: Union[Literal["odc.stac", "stackstac"], None] = None, |
| 14 | + stacking_library: Literal["odc.stac", "stackstac"] | None = None, |
14 | 15 | **kwargs, |
15 | 16 | ) -> xarray.Dataset: |
16 | 17 | """Given a PySTAC object return an xarray dataset. |
@@ -43,9 +44,9 @@ def to_xarray( |
43 | 44 | @to_xarray.register(pystac.Item) |
44 | 45 | @to_xarray.register(pystac.ItemCollection) |
45 | 46 | def _( |
46 | | - obj: Union[pystac.Item, pystac.ItemCollection], |
47 | | - drop_variables: Union[str, List[str], None] = None, |
48 | | - stacking_library: Union[Literal["odc.stac", "stackstac"], None] = None, |
| 47 | + obj: pystac.Item | pystac.ItemCollection, |
| 48 | + drop_variables: str | list[str] | None = None, |
| 49 | + stacking_library: Literal["odc.stac", "stackstac"] | None = None, |
49 | 50 | **kwargs, |
50 | 51 | ) -> xarray.Dataset: |
51 | 52 | if drop_variables is not None: |
|
86 | 87 | @to_xarray.register |
87 | 88 | def _( |
88 | 89 | obj: pystac.Asset, |
89 | | - stacking_library: Union[Literal["odc.stac", "stackstac"], None] = None, |
| 90 | + stacking_library: Literal["odc.stac", "stackstac"] | None = None, |
90 | 91 | **kwargs, |
91 | 92 | ) -> xarray.Dataset: |
92 | 93 | default_kwargs: Mapping = {"chunks": {}} |
|
0 commit comments