U Å9%e<ã@stUddlmZmZmZddlmZGdd„dƒZiaeejefe d<ejeddœdd„Z ejeed œd d „Z dS) é)ÚcastÚDictÚOptionalNc@s eZdZdS)Ú_StateN)Ú__name__Ú __module__Ú __qualname__©r r úb/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/torch/distributed/_composable_state.pyrsrÚ_module_state_mapping)ÚmoduleÚstateÚreturncCs$|tkstd|›dƒ‚|t|<dS)Nz Inserting z more than once.)r ÚAssertionError)r r r r r Ú_insert_module_state sr)r rcCs,t|tƒrtt|ƒS|tkr$t|SdSdS)aK 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 )r r r r Ú_get_module_states  r) ÚtypingrrrZtorch.nnÚnnrr ÚModuleÚ__annotations__rrr r r r Ús