a jDf@s6UddlmZddlmZddlmZddlmZmZm Z ddl Z ddl m Z ddlmZddlmZmZee jfZere d Zeeed fZdd lmZmZmZGd d d ZiZde d<dd dddZ!dddddZ"dddddZ#dddddZ$ddddd Z%dd!d"d#d$d%Z&d&d!d'd#d(d)Z'dS)*) annotations) import_module) ModuleType) TYPE_CHECKINGAnyLiteralN)Version) is_scalar) is_duck_arrayis_duck_dask_array)daskpintcupysparsecubednumbagg.)_DType _ShapeType duckarrayc@sBeZdZUdZded<ded<ded<ded <d d d d dZdS)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 | NonemodulerversionDuckArrayTypestypebool availableModTypeNonemodreturnc Cszt|}t|j}|dkr*tdjf}nZ|dkr<|jf}nH|dkrN|jf}n6|dkr`|jf}n$|dkrr|jf}n|dkrd}ntWn&tt fyd}td }d}Yn0||_ ||_ ||_ |du|_ dS) Nr z dask.arrayr rrrrz0.0.0)rr __version__ArrayZQuantityZndarrayZ SparseArrayNotImplementedError ImportErrorAttributeErrorrrrr)selfrZduck_array_moduleZduck_array_versionZduck_array_typer!r!g/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/xarray/namedarray/pycompat.py__init__"s0      zDuckArrayModule.__init__N)__name__ __module__ __qualname____doc____annotations__r)r!r!r!r(rs rzdict[ModType, DuckArrayModule]_cached_duck_array_modulesrrcCs(|tvrt|}|t|<|St|SdSN)r/r)rZduckmodr!r!r(_get_cached_duck_array_moduleHs r1rcCs t|jS)z3Quick wrapper to get the array class of the module.)r1rrr!r!r( array_typeQsr3rcCs t|jS)z/Quick wrapper to get the version of the module.)r1rr2r!r!r( mod_versionVsr4zduckarray[Any, Any]r)xr cCst|pt|ot|dS)Nchunks)r r hasattrr5r!r!r(is_chunked_array[sr9cCst|ot|Sr0)r r r8r!r!r(is_0d_dask_array_sr:zdict[str, Any]znp.ndarray[Any, np.dtype[Any]])datakwargsr cKsddlm}ddlm}t||r*|}t|rP||}|j|fi|^}}t|tdrf| }t|tdrz|j }t|tdr| }t |}|S)NrExplicitlyIndexedget_chunked_array_typerr r)xarray.core.indexingr> xarray.namedarray.parallelcompatr@ isinstanceget_duck_arrayr9computer3getZ magnitudeZtodensenpasarray)r;r<r>r@ chunkmanager_r!r!r(to_numpycs    rKrzduckarray[_ShapeType, _DType]cKsnddlm}ddlm}t|rB||}|j|fi|^}}|St||rT|St|r`|St |SdS)Nrr=r?) rAr>rBr@r9rErCrDr rGrH)r;r<r>r@rIZ loaded_datarJr!r!r( to_duck_array|s   rL)( __future__r importlibrtypesrtypingrrrnumpyrGZpackaging.versionrZxarray.core.utilsr Zxarray.namedarray.utilsr r intinteger integer_typesrtuplerrZxarray.namedarray._typingrrrrr/r.r1r3r4r9r:rKrLr!r!r!r(s*     0