d$dZddlZdZdZdZdZdZdZd Zd Z d Z d Z d Z dZ dZdZdZddiZddiZdZdZdZdZdZdZdZdZdZdZdZd7d Zd!ZiZ eeezed"ze d#<eeezed$ze d%<eeezed&ze d'<eee d(<eee d)<d*Z!d+Z"d,Z#d-Z$d.Z%d/Z&e'd0krse(ej)d1krej)d2*ndZ+e+d3vre,d4ej)dd5e+d6kZ-e%e e-D]Z.e/e.dSdS)8aGenerate module and stub file for arithmetic operators of various xarray classes. For internal xarray development use only. Usage: python xarray/util/generate_ops.py --module > xarray/core/_typed_ops.py python xarray/util/generate_ops.py --stubs > xarray/core/_typed_ops.pyi N))__eq__znputils.array_eq)__ne__znputils.array_ne))__lt__z operator.lt)__le__z operator.le)__gt__z operator.gt)__ge__z operator.ge) )__add__ operator.add)__sub__ operator.sub)__mul__ operator.mul)__pow__ operator.pow) __truediv__operator.truediv) __floordiv__operator.floordiv)__mod__ operator.mod)__and__ operator.and_)__xor__ operator.xor)__or__ operator.or_) __lshift__zoperator.lshift) __rshift__zoperator.rshift) )__radd__r )__rsub__r )__rmul__r)__rpow__r) __rtruediv__r) __rfloordiv__r)__rmod__r)__rand__r)__rxor__r)__ror__r) )__iadd__z operator.iadd)__isub__z operator.isub)__imul__z operator.imul)__ipow__z operator.ipow) __itruediv__zoperator.itruediv) __ifloordiv__zoperator.ifloordiv)__imod__z operator.imod)__iand__z operator.iand)__ixor__z operator.ixor)__ior__z operator.ior) __ilshift__zoperator.ilshift) __irshift__zoperator.irshift))__neg__z operator.neg)__pos__z operator.pos)__abs__z operator.abs) __invert__zoperator.invert))roundz ops.round_)argsortz ops.argsort)conjzops.conj) conjugatez ops.conjugatezM def {method}(self, other): return self._binary_op(other, {func})z] def {method}(self, other): return self._binary_op(other, {func}, reflexive=True)zU def {method}(self, other): return self._inplace_binary_op(other, {func})z> def {method}(self): return self._unary_op({func})z` def {method}(self, *args, **kwargs): return self._unary_op({func}, *args, **kwargs)zO def _unary_op(self, f, *args, **kwargs): raise NotImplementedErrorzW def _binary_op(self, other, f, reflexive=False): raise NotImplementedErrorzN def _inplace_binary_op(self, other, f): raise NotImplementedErroroverridez # type: ignore[override]zR def {method}(self: T_Dataset, other: DsCompatible) -> T_Dataset: ...{override}z @overload{override} def {method}(self, other: T_Dataset) -> T_Dataset: ... @overload def {method}(self, other: "DatasetGroupBy") -> "Dataset": ... @overload def {method}(self: T_DataArray, other: DaCompatible) -> T_DataArray: ...z @overload{override} def {method}(self, other: T_Dataset) -> T_Dataset: ... @overload def {method}(self, other: T_DataArray) -> T_DataArray: ... @overload def {method}(self: T_Variable, other: VarCompatible) -> T_Variable: ...z @overload{override} def {method}(self, other: T_Dataset) -> T_Dataset: ... @overload def {method}(self, other: "DataArray") -> "Dataset": ... @overload def {method}(self, other: GroupByIncompatible) -> NoReturn: ...z @overload{override} def {method}(self, other: T_Dataset) -> T_Dataset: ... @overload def {method}(self, other: T_DataArray) -> T_DataArray: ... @overload def {method}(self, other: GroupByIncompatible) -> NoReturn: ...z7 def {method}(self: {self_type}) -> {self_type}: ...zH def {method}(self: {self_type}, *args, **kwargs) -> {self_type}: ...z0 def _unary_op(self, f, *args, **kwargs): ...z6 def _binary_op(self, other, f, reflexive=...): ...z/ def _inplace_binary_op(self, other, f): ...c~d|i}dgttifttt|ft t t|fgS)N self_typeNN)required_method_unarystub_required_unary UNARY_OPStemplate_unary stub_unaryOTHER_UNARY_METHODStemplate_other_unarystub_other_unary)r@ extra_contexts 8lib/python3.11/site-packages/xarray/util/generate_ops.pyunopsrLs@ ),M .0CRH NJ > 24DmT cdgttifttzt|t ft t|tftt|t fgS)NrA) required_method_binarystub_required_binary BINOPS_NUM BINOPS_CMPtemplate_binop NO_OVERRIDE BINOPS_EQNEOVERRIDE_TYPESHEDBINOPS_REFLEXIVEtemplate_reflexive)stubs rKbinopsrZsG /1ErJ j .$ D nd,=> -t[A  rMcDdgttifttdifgS)NrAr>)required_method_inplacestub_required_inplaceBINOPS_INPLACEtemplate_inplacerMrKinplaceras) 02GL )2r2 rM T_DatasetDatasetOpsMixin T_DataArrayDataArrayOpsMixin T_VariableVariableOpsMixinDatasetGroupByOpsMixinDataArrayGroupByOpsMixinz"""Mixin classes with arithmetic operators.""" # This file was generated using xarray.util.generate_ops. Do not edit manually. import operator from . import nputils, opsaK"""Stub file for mixin classes with arithmetic operators.""" # This file was generated using xarray.util.generate_ops. Do not edit manually. from typing import NoReturn, TypeVar, overload import numpy as np from numpy.typing import ArrayLike from .dataarray import DataArray from .dataset import Dataset from .groupby import DataArrayGroupBy, DatasetGroupBy, GroupBy from .types import ( DaCompatible, DsCompatible, GroupByIncompatible, ScalarOrArray, VarCompatible, ) from .variable import Variable try: from dask.array import Array as DaskArray except ImportError: DaskArray = np.ndarray # type: ignore # DatasetOpsMixin etc. are parent classes of Dataset etc. # Because of https://github.com/pydata/xarray/issues/5755, we redefine these. Generally # we use the ones in `types`. (We're open to refining this, and potentially integrating # the `py` & `pyi` files to simplify them.) T_Dataset = TypeVar("T_Dataset", bound="DatasetOpsMixin") T_DataArray = TypeVar("T_DataArray", bound="DataArrayOpsMixin") T_Variable = TypeVar("T_Variable", bound="VariableOpsMixin")z.{newline} class {cls_name}: __slots__ = ()z% {method}.__doc__ = {func}.__doc__c#K|rtntV|D]<\}}t|d|zVt ||Ed{V=dS)zRender the module or stub file. )cls_namenewlineN)MODULE_PREAMBLESTUBFILE_PREAMBLEitemsCLASS_PREAMBLEformat_render_classbody)ops_info is_modulerl method_blockss rKrenderrws& =//,====#+>>#3#3??-##Xti?O#PPPPP$]I>>>>>>>>>>??rMc#K|D])\}}}}|r|n|}|r|D]\}}|jd||d|V*|r5dV|D]0^}} |D]'\}}|r |rt||V(/dSdS)N)methodfuncr>r`)rrCOPY_DOCSTRING) rvrumethod_func_pairsmethod_template stub_templateextratemplateryrz_s rKrsrssDQII@?M5&/B??]  I 1 I I %hoHV$HH%HHHHHHJ%2 J J !  1 J J JdJ(//vD/IIIII JJJ J JrM__main__>--stubs--modulezUsage: z --module | --stubsr)r>)0__doc__sysrUrRrQrWr^rDrGrSrXr_rErHrBrOr\rVrTstub_dsstub_dastub_var stub_dsgb stub_dagbrFrIrCrPr]rLrZrartrnrorqr{rwrs__name__lenargvloweroption SystemExitrulineprintr`rMrKrs N       1A9):%%%  !=>2  V P O G  G ; L4:3 $fWoo 9EE+R>R R %vh//''));eeL>Q>QQ %+VI%6%6 !"'-vi'8'8 #$ @F)??? J J J z$'CMMQ$6$6SXa[   DF ,,,jC38A;CCCDDD*$Ix++ d  rM