extra_label

tosh2230 · Oct 26, 2024 · c0f93b1 · c0f93b1
1 parent c9962d6
commit c0f93b1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/stairlight/map.py b/src/stairlight/map.py
@@ -155,14 +155,14 @@ def remap(
                 ),
             )
 
-            upstairs_extra_labels = [
+            upstairs_extra_labels: list[dict[str, Any]] = [
                 extra_label.get(MappingConfigKey.LABELS, {})
                 for extra_label in extra_labels
                 if extra_label.get(MappingConfigKey.TABLE_NAME)
                 == upstair_table_reference.TableName
             ]
             upstairs_extra_label = (
-                upstairs_extra_labels[0] if upstairs_extra_labels else []
+                upstairs_extra_labels[0] if upstairs_extra_labels else {}
             )
             upstair_template = self.create_upstair_template(
                 template=template,
@@ -237,7 +237,7 @@ def create_upstair_template(
         Args:
             template (Template): Template class
             mapped_labels (dict[str, Any]): Labels in mapping section
-            extra_labels (list[dict[str, Any]]): Extra labels
+            extra_label (dict[str, Any]): Extra labels
             upstairs (str): Upstairs table's Name
 
         Returns: