Fix typing issues

youtype · Apr 24, 2024 · bd027f8 · bd027f8
1 parent 4410431
commit bd027f8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mypy_boto3_builder/import_helpers/import_record.py b/mypy_boto3_builder/import_helpers/import_record.py
@@ -31,7 +31,7 @@ class ImportRecord:
     )
 
     def __init__(
-        self: Self,
+        self,
         source: ImportString,
         name: str = "",
         alias: str = "",

diff --git a/mypy_boto3_builder/parsers/shape_parser.py b/mypy_boto3_builder/parsers/shape_parser.py
@@ -181,7 +181,7 @@ def _get_resource_names(self) -> list[str]:
         return list(self._resources_shape["resources"].keys())
 
     def _get_resource_shape(self, name: str) -> ResourceShape:
-        if not self._resources_shape:
+        if not self._resources_shape or "resources" not in self._resources_shape:
             raise ShapeParserError("Resource shape not found")
         try:
             return self._resources_shape["resources"][name]