🐛 (transform) ensure default dataset for mapping

investigativedata · Oct 6, 2023 · 50d0300 · 50d0300
1 parent 190406e
commit 50d0300
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/investigraph/logic/transform.py b/investigraph/logic/transform.py
@@ -14,7 +14,9 @@
 from investigraph.types import CEGenerator, SDict
 
 
-def map_record(record: SDict, mapping: QueryMapping, dataset: str) -> CEGenerator:
+def map_record(
+    record: SDict, mapping: QueryMapping, dataset: str | None = "default"
+) -> CEGenerator:
     mapping = mapping.get_mapping()
     if mapping.source.check_filters(record):
         entities = mapping.map(record)