small change

dlt-hub · Oct 11, 2023 · 37ac216 · 37ac216
1 parent 3914457
commit 37ac216
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 15 deletions.
diff --git a/dlt/common/normalizers/json/relational.py b/dlt/common/normalizers/json/relational.py
@@ -30,7 +30,7 @@ class TDataItemRowChild(TDataItemRow, total=False):
     value: Any  # for lists of simple types
 
 
-class RelationalNormalizerConfigPropagation(TypedDict, total=True):
+class RelationalNormalizerConfigPropagation(TypedDict, total=False):
     root: Optional[Mapping[str, TColumnName]]
     tables: Optional[Mapping[str, Mapping[str, TColumnName]]]
 
@@ -261,16 +261,12 @@ def extend_schema(self) -> None:
         # for every table with the write disposition merge, we propagate the root_key
         for table_name, table in self.schema.tables.items():
             if not table.get("parent") and table["write_disposition"] == "merge":
-                prop_config: RelationalNormalizerConfigPropagation = {
-                    "root": {
-                    },
+                DataItemNormalizer.update_normalizer_config(self.schema, {"propagation": {
                     "tables": {
-                    table_name: {
+                        table_name: {
                             "_dlt_id": TColumnName("_dlt_root_id")
                         }
-                    }
-                }
-                DataItemNormalizer.update_normalizer_config(self.schema, {"propagation": prop_config})
+                    }}})
 
     def normalize_data_item(self, item: TDataItem, load_id: str, table_name: str) -> TNormalizedRowIterator:
         # wrap items that are not dictionaries in dictionary, otherwise they cannot be processed by the JSON normalizer

diff --git a/dlt/extract/source.py b/dlt/extract/source.py
@@ -697,13 +697,11 @@ def root_key(self) -> bool:
     @root_key.setter
     def root_key(self, value: bool) -> None:
         if value is True:
-            propagation_config: RelationalNormalizerConfigPropagation = {
-                "root": {
-                    "_dlt_id": TColumnName("_dlt_root_id")
-                },
-                "tables": {}
-            }
-            RelationalNormalizer.update_normalizer_config(self._schema, {"propagation": propagation_config})
+            RelationalNormalizer.update_normalizer_config(self._schema,
+                {"propagation": {
+                    "root": {
+                        "_dlt_id": TColumnName("_dlt_root_id")
+                    }}})
         else:
             if self.root_key:
                 propagation_config = RelationalNormalizer.get_normalizer_config(self._schema)["propagation"]