{".class":"MypyFile","_fullname":"torch.distributed._composable_state","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","Dict":{".class":"SymbolTableNode","cross_ref":"typing.Dict","kind":"Gdef"},"Optional":{".class":"SymbolTableNode","cross_ref":"typing.Optional","kind":"Gdef"},"_State":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["builtins.object"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch.distributed._composable_state._State","name":"_State","type_vars":[]},"deletable_attributes":[],"flags":[],"fullname":"torch.distributed._composable_state._State","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch.distributed._composable_state","mro":["torch.distributed._composable_state._State","builtins.object"],"names":{".class":"SymbolTable"},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch.distributed._composable_state._State.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch.distributed._composable_state._State","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch.distributed._composable_state.__annotations__","name":"__annotations__","type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch.distributed._composable_state.__doc__","name":"__doc__","type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch.distributed._composable_state.__file__","name":"__file__","type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch.distributed._composable_state.__name__","name":"__name__","type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch.distributed._composable_state.__package__","name":"__package__","type":"builtins.str"}},"_get_module_state":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["module"],"dataclass_transform_spec":null,"flags":[],"fullname":"torch.distributed._composable_state._get_module_state","name":"_get_module_state","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["module"],"arg_types":["torch.nn.modules.module.Module"],"bound_args":[],"def_extras":{"first_arg":null},"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_ellipsis_args":false,"name":"_get_module_state","ret_type":{".class":"UnionType","items":["torch.distributed._composable_state._State",{".class":"NoneType"}]},"type_guard":null,"unpack_kwargs":false,"variables":[]}}},"_insert_module_state":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["module","state"],"dataclass_transform_spec":null,"flags":[],"fullname":"torch.distributed._composable_state._insert_module_state","name":"_insert_module_state","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["module","state"],"arg_types":["torch.nn.modules.module.Module","torch.distributed._composable_state._State"],"bound_args":[],"def_extras":{"first_arg":null},"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_ellipsis_args":false,"name":"_insert_module_state","ret_type":{".class":"NoneType"},"type_guard":null,"unpack_kwargs":false,"variables":[]}}},"_module_state_mapping":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","has_explicit_value"],"fullname":"torch.distributed._composable_state._module_state_mapping","name":"_module_state_mapping","type":{".class":"Instance","args":["torch.nn.modules.module.Module","torch.distributed._composable_state._State"],"type_ref":"builtins.dict"}}},"cast":{".class":"SymbolTableNode","cross_ref":"typing.cast","kind":"Gdef"},"nn":{".class":"SymbolTableNode","cross_ref":"torch.nn","kind":"Gdef"}},"path":"/home/bigcatc/miniconda3/envs/py310/lib/python3.10/site-packages/torch/distributed/_composable_state.py"}