uhh \UdZddlmZmZddlmZmZmZmZddl Z gZ e e ed<dee e ffdZde e fdZde fd Zd Zeed <eZeed <eZeed <eZeed< erddlmZddlZddlmZddlmZmZm Z ddl!m"Z"m#Z#ee"gej$j%fZ&eed<de"dej$j%fdZ'dee de de(fdZ)dede de e*e e dzfddfdZ+Gdde#Z,de de-fd Z.dS#e/$rZ0de de-fd!Z.YdZ0[0dSdZ0[0wwxYw)"aA mypy_ plugin for managing a number of platform-specific annotations. Its functionality can be split into three distinct parts: * Assigning the (platform-dependent) precisions of certain `~numpy.number` subclasses, including the likes of `~numpy.int_`, `~numpy.intp` and `~numpy.longlong`. See the documentation on :ref:`scalar types ` for a comprehensive overview of the affected classes. Without the plugin the precision of all relevant classes will be inferred as `~typing.Any`. * Removing all extended-precision `~numpy.number` subclasses that are unavailable for the platform in question. Most notably this includes the likes of `~numpy.float128` and `~numpy.complex256`. Without the plugin *all* extended-precision types will, as far as mypy is concerned, be available to all platforms. * Assigning the (platform-dependent) precision of `~numpy.ctypeslib.c_intp`. Without the plugin the type will default to `ctypes.c_int64`. .. versionadded:: 1.22 .. deprecated:: 2.3 Examples -------- To enable the plugin, one must add it to their mypy `configuration file`_: .. code-block:: ini [mypy] plugins = numpy.typing.mypy_plugin .. _mypy: https://mypy-lang.org/ .. _configuration file: https://mypy.readthedocs.io/en/stable/config_file.html )CallableIterable) TYPE_CHECKINGFinal TypeAliascastN__all__returnc dtjfdtjfdtjfdtjfdtjfdtjfdtjfdtjfd tj fd tj fd tj fg }i}|D];\}}d tj |j z}td |d|td|<<|S)N _NBitByte _NBitShort _NBitIntC _NBitIntP_NBitInt _NBitLong _NBitLongLong _NBitHalf _NBitSingle _NBitDouble_NBitLongDoublez ._nbit_base._Bit._nbit.)npbyteshortintcintpint_longlonglonghalfsingledouble longdoubledtypeitemsize_MODULE)namesretnametypns O/var/www/histauto/venv/lib/python3.11/site-packages/numpy/typing/mypy_plugin.py_get_precision_dictr/,s bg rx bg bg RW bg "+& bg  "  " BM* ECII c  & &,3)H)H!)H)H)Hw % %t % %&& Jc"gd}d|DS)N)float96float128 complex192 complex256c<g|]}tt||S)hasattrr).0is r. z0_get_extended_precision_list..Is% 8 8 8!Q 8A 8 8 8r0r7)extended_namess r._get_extended_precision_listr=Bs)N 9 8~ 8 8 88r0cdddddtjdjdS)Nc_intc_long c_longlong)r:lqr-)getrr&charr7r0r._get_c_intp_namerFKs8     c"(3-- h'' (r0z numpy._typingr(_PRECISION_DICT_EXTENDED_PRECISION_LIST_C_INTP) TypeAnalyser)PRI_MED) ImportFromMypyFile Statement)AnalyzeTypeContextPlugin _HookFuncctxc|\}}}|jdd}ttd|}t d||S)z;Replace a type-alias with a concrete ``NBitBase`` subclass..rrJ)r+splitrGr(r named_type)rRr,_apir+name_news r._hookr[ss[ Qx~~c""2&"g#<#Identify the first ``ImportFrom`` instance the specified `id`.r]NzBFailed to identify a `ImportFrom` instance with the following id: ) enumerategetattr ValueError)r\r]r:values r._indexrczsf!(++  HAuudD))R//0:35::;; ;r0filemoduleimportsct|d|}d|_|jtd|jfD]}t ||}|||<dS)zA mypy plugin for handling versus numpy-specific typing tasks.fullnamer Nc&|tvrtSdS)zSet the precision of platform-specific `numpy.number` subclasses. For example: `numpy.int_`, `numpy.longlong` and `numpy.longdouble`. N)rGr[)selfros r.get_type_analyze_hookz"_NumpyPlugin.get_type_analyze_hooks ?** 4r0rdc|j}|dkr*t|tddtDn |dkrt|dtdfgt |dfgS) a Handle all import-based overrides. * Import platform-specific extended-precision `numpy.number` subclasses (*e.g.* `numpy.float96` and `numpy.float128`). * Import the appropriate `ctypes` equivalent to `numpy.intp`. numpyz._extended_precisioncg|]}||fSr7r7)r9vs r.r;z4_NumpyPlugin.get_additional_deps..sFFFaVFFFr0)rfznumpy.ctypeslibctypes_c_intprU)rorlr(rHrIrK)rqrdros r.get_additional_depsz _NumpyPlugin.get_additional_depss}H7""!444FF-EFFF ...!%y12 h+, ,r0) __name__ __module__ __qualname____doc__strrQrrrMlisttupleintryr7r0r.rnrnsuLL # )d:J     -  - %S# & ' - - - - - -r0rnversionchddl}d}d|d|d}||tdtS)Nrznumpy.typing.mypy_plugin`zS` is deprecated, and will be removed in a future release. Please remove `plugins = z/` in your mypy config.(deprecated in NumPy 2.3)) stacklevel)warningswarnDeprecationWarningrn)rrpluginwarn_msgs r.rrsY+ ) ) )17 ) ) )   h 2q AAAr0ct)N)e)rs r.rrlsr0)1r}collections.abcrrtypingrrrrrtrr rr~__annotations__dictr/r=rFr(rGrHrI mypy.typeanalrJ mypy.typesmypy mypy.buildrK mypy.nodesrLrMrN mypy.pluginrOrPtypesTyperQr[rrcrrlrntyperModuleNotFoundErrorrr7r0r.rs$!!!F/.......888888888888cT#s(^,9d3i9999(#((((! -,.....#?">"@"@%@@@"!#####b/......""""""::::::::::66666666$%7$8$*/$IJIyJJJ>%>$*/>>>>;),;#;#;;;;      eCtO,-        $-$-$-$-$-v$-$-$-L        [s>$DD+D&&D+