+ j{aR t 0t^RIt^RIHt^RIHt!RR4t]P!4s]^kRRlt RRlt R#) N)castc]tRt^tRtR#)_StateN)__name__ __module__ __qualname____firstlineno____static_attributes__rd/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/distributed/_composable_state.pyrrsr rcHV^8dQhR\PR\RR/#)modulestatereturnNnnModuler)formats"r __annotate__rs%77767d7r ctV\9d\RV R24h\P!V4\V&R#)z Inserting z more than once.N)_module_state_mappingAssertionErrorweakrefref)rrs&&r _insert_module_staters4 &&z&1ABCC$+KK$6&!r cRV^8dQhR\PR\R,/#)rrrNr)rs"r rrs"biiFTMr c\V\4'd\\V4#V\9d&\V,pV!4pVf \ R4hV#R#)aX Return the ``_State`` in ``model``. Given a ``module``, this API finds out if the module is also a ``_State`` instance or if the module is managed by a composable API. If the module is also a ``_State``, ``module`` will be casted to ``_State` and returned. If it is managed by a composable API, the corresponding ``_State`` will be returned. Nz(State has already been garbage collected) isinstancerrrr)r state_refrs& r _get_module_stater"sS&&!!FF## * *-f5IKE}$%OPPLr cV^8dQh/^\9dD\P\P\P \ ,3,;R&#)rr)__conditional_annotations__rWeakKeyDictionaryrr ReferenceTyper)rs"r rrs>  w00IIw$$V,, r ) r$rtypingrtorch.nnrrr%rrr"r)r$s@r r)s=    7r