| 
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