dW UddlmZddlmZddlmZddlmZmZm Z ddl Z ddl m Z ddlmZmZmZee jfZere dZeeed fZGd d ZiZd ed <ddZddZddZdZdZ ddZ!dZ"dS)) annotations) import_module) ModuleType) TYPE_CHECKINGAnyLiteralN)Version) is_duck_array is_scalarmodule_available)daskpintcupysparsecubed.cDeZdZUdZded<ded<ded<ded <ddZdS)DuckArrayModulez Solely for internal isinstance and version checks. Motivated by having to only import pint when required (as pint currently imports xarray) https://github.com/pydata/xarray/pull/5561#discussion_r664815718 zModuleType | Nonemoduler versionDuckArrayTypestypebool availablemodModTypereturnNonec t|}t|j}|dkrtdjf}nC|dkr |jf}n4|dkr |jf}n%|dkr |jf}n|dkr |jf}ntn*#ttf$rd}td}d}YnwxYw||_ ||_ ||_ |du|_ dS) Nr z dask.arrayrrrrz0.0.0)rr __version__ArrayQuantityndarray SparseArrayNotImplementedError ImportErrorAttributeErrorrrrr)selfrduck_array_moduleduck_array_versionduck_array_types 4lib/python3.11/site-packages/xarray/core/pycompat.py__init__zDuckArrayModule.__init__ s ! -c 2 2 !():)F!G!G f}}#0#>#>#D"F#4#="?#4#<">#4#@"B#4#:"<))^, ! ! ! $ !(!1!1  OOO ! ( ) # *$6sBB$B,+B,N)rrrr)__name__ __module__ __qualname____doc____annotations__r-rr,rrscOOO777777r3rzdict[ModType, DuckArrayModule]_cached_duck_array_modulesrrrcd|tvrt|}|t|<|St|SN)r4r)rduckmods r,_get_cached_duck_array_moduler8Cs6 ,,,!#&&*1"3')#..r3rc*t|jS)z3Quick wrapper to get the array class of the module.)r8rrs r, array_typer;Ls ( - - 22r3r c*t|jS)z/Quick wrapper to get the version of the module.)r8rr:s r, mod_versionr=Qs ( - - 55r3cFtdrddlm}||SdS)Nr r)is_dask_collectionF)r dask.baser?)xr?s r,r?r?Vs:%000000!!!$$$ 5r3c>t|ot|Sr6)r r?rAs r,is_duck_dask_arrayrD^s    5 21 5 55r3rc^t|pt|ot|dS)Nchunks)rDr hasattrrCs r,is_chunked_arrayrHbs, a O]1%5%5%N'!X:N:NOr3c>t|ot|Sr6)rDr rCs r,is_0d_dask_arrayrJfs a 1Yq\\1r3)rrrr)rrrr)rrrr )rr)# __future__r importlibrtypesrtypingrrrnumpynppackaging.versionr xarray.core.utilsr r r intinteger integer_typesrtuplerrrr4r2r8r;r=r?rDrHrJrr3r,rWs"""""""######..........%%%%%%HHHHHHHHHHbj! +?@G49c>*N*7*7*7*7*7*7*7*7Z>@????////3333 6666 666PPPP22222r3