ȅi{%SSKrSSKJr SSKJr "SS5r\R "5q\R \R\R\4\ S'S\RS\SS4S jr S\RS\S-4S jr g) N)castc\rSrSrSrg)_StateN)__name__ __module__ __qualname____firstlineno____static_attributes__r]/home/james-whalen/.local/lib/python3.13/site-packages/torch/distributed/_composable_state.pyrrsr r_module_state_mappingmodulestatereturncpU[;a[SUS35e[R"U5[U'g)Nz Inserting z more than once.)rAssertionErrorweakrefref)rrs r_insert_module_staters4 &&z&1ABCC$+KK$6&!r c[U[5(a[[U5$U[;a [UnU"5nUc [ S5eU$g)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_statersS&&!!FF## * *-f5IKE}$%OPPLr ) rtypingrtorch.nnnnrWeakKeyDictionaryrModule ReferenceType__annotations__rrrr rr#s   w00IIw$$V,, 7767d7biiFTMr