add return type

chanzuckerberg · Nov 18, 2023 · 45c7f34 · 45c7f34
1 parent 1e6d9e0
commit 45c7f34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/compute_mappings/compute_ancestor_mapping.py b/scripts/compute_mappings/compute_ancestor_mapping.py
@@ -42,7 +42,7 @@ def recurse(x):  # type: ignore
 # In[ ]:
 
 
-def create_ancestors_mapping(onto, classes, prefix=None):  # type: ignore
+def create_ancestors_mapping(onto, classes, prefix=None) -> Dict[str, List[str]]:
     ancestors: Dict[str, List[str]] = defaultdict(list)
     for cls in classes:
         class_key = cls.name.replace("_", ":")