Add type hint for Model.get_attributes (pynamodb#589)

ikonst · web-flow · commit 4462874d224a · 2019-01-29T23:02:10.000-08:00
diff --git a/pynamodb/models.pyi b/pynamodb/models.pyi
@@ -1,3 +1,4 @@
+from .attributes import Attribute
 from .exceptions import DoesNotExist as DoesNotExist
 from typing import Any, Dict, Generic, Iterable, Iterator, List, Optional, Sequence, Tuple, Type, TypeVar, Text, Union
 
@@ -125,7 +126,9 @@ class Model(metaclass=MetaModel):
     @classmethod
     def get_throttle(cls): ...
     @classmethod
-    def _get_attributes(cls) -> Dict[str, Any]: ...
+    def get_attributes(cls) -> Dict[str, Attribute]: ...
+    @classmethod
+    def _get_attributes(cls) -> Dict[str, Attribute]: ...
     @classmethod
     def _get_connection(cls) -> TableConnection: ...
 
