From 960e8c898dc0826d6dd5991f087fcd9d3ee6fcd1 Mon Sep 17 00:00:00 2001 From: Melody Horn Date: Wed, 23 Dec 2020 00:53:58 -0700 Subject: fix some typing edge cases --- crowbar_reference_compiler/ast.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'crowbar_reference_compiler') diff --git a/crowbar_reference_compiler/ast.py b/crowbar_reference_compiler/ast.py index 37ce0da..142fa8e 100644 --- a/crowbar_reference_compiler/ast.py +++ b/crowbar_reference_compiler/ast.py @@ -347,7 +347,7 @@ class HeaderFile: def get_declarations(self) -> List[Declaration]: included_declarations = [x.get_declarations() for x in self.includes] - own_declarations = [x for x in self.contents if isinstance(x, Declaration)] + own_declarations: List[Declaration] = [x for x in self.contents if isinstance(x, Declaration)] all_declarations = included_declarations + [own_declarations] return [x for l in all_declarations for x in l] @@ -359,7 +359,7 @@ class ImplementationFile: def get_declarations(self) -> List[Declaration]: included_declarations = [x.get_declarations() for x in self.includes] - own_declarations = [x for x in self.contents if isinstance(x, Declaration)] + own_declarations: List[Declaration] = [x for x in self.contents if isinstance(x, Declaration)] all_declarations = included_declarations + [own_declarations] return [x for l in all_declarations for x in l] -- cgit v1.2.3