קgdUddlmZmZmZddlmZGddZiaeejefe d<dejdeddfd Z dejdeefd Z dS) )castDictOptionalNceZdZdS)_StateN)__name__ __module__ __qualname___/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/distributed/_composable_state.pyrrsDr r_module_state_mappingmodulestatereturncH|tvsJd|d|t|<dS)Nz Inserting z more than once.)r)rrs r _insert_module_stater s9 . . . .0UV0U0U0U . . .$)&!!!r ct|trtt|S|tvr t|SdS)at 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. N) isinstancerrr)rs r _get_module_statersD&&!!FF### * * *(0 04r ) typingrrrtorch.nnnnrrModule__annotations__rrr r r rs'''''''''''        24tBIv-.333**6*d**** biHV,<r