Mark nested output shapes as total

youtype · Apr 29, 2024 · ccdff2e · ccdff2e
1 parent 01e3bb8
commit ccdff2e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mypy_boto3_builder/parsers/shape_parser.py b/mypy_boto3_builder/parsers/shape_parser.py
@@ -403,8 +403,9 @@ def _parse_shape_structure(
                 ),
                 attr_name in required,
             )
-        if output:
+        if is_output_or_child:
             self._mark_typed_dict_as_total(typed_dict)
+        if output:
             self._add_response_metadata(typed_dict)
 
         if found_typed_dict and not typed_dict.is_same(found_typed_dict):