File tree Expand file tree Collapse file tree 2 files changed +6
-1
lines changed Expand file tree Collapse file tree 2 files changed +6
-1
lines changed Original file line number Diff line number Diff line change @@ -331,7 +331,7 @@ def get_keystr(key_path: KeyPath) -> str:
331
331
return f"*args{ keystr (key_path [1 :])} "
332
332
else :
333
333
kwarg_key = key_path [1 ]
334
- assert isinstance (kwarg_key , MappingKey )
334
+ assert isinstance (kwarg_key , GetAttrKey )
335
335
name = str (kwarg_key )[1 :- 1 ] # get rid of the enclosed []
336
336
return f"{ name } { keystr (key_path [2 :])} "
337
337
Original file line number Diff line number Diff line change @@ -134,6 +134,11 @@ class _SubmoduleBase:
134
134
_ty : Optional [str ]
135
135
136
136
def type_name (self ) -> Optional [str ]:
137
+ """
138
+ Subclass of this class - InterpreterModule, InterpreterModuleDispatcher, represents
139
+ corresponding model in eager model. To get this type information for those modules
140
+ in eager model we need to use this method.
141
+ """
137
142
return self ._ty
138
143
139
144
You can’t perform that action at this time.
0 commit comments