diff --git a/docforge/model/object.py b/docforge/model/object.py index 9181fe5..8edb79c 100644 --- a/docforge/model/object.py +++ b/docforge/model/object.py @@ -17,13 +17,13 @@ class DocObject: self.path = path self.signature = signature self.docstring = docstring - self.members: Dict[str, DocObject] = {} + self.members: Dict[str, 'DocObject'] = {} - def add_member(self, obj: DocObject) -> None: + def add_member(self, obj: 'DocObject') -> None: self.members[obj.name] = obj - def get_member(self, name: str) -> DocObject: + def get_member(self, name: str) -> 'DocObject': return self.members[name] - def get_all_members(self) -> Iterable[DocObject]: + def get_all_members(self) -> Iterable['DocObject']: return self.members.values()