From 476a9d00ebb8ffb67a7a5a1e0564510f259fd3c2 Mon Sep 17 00:00:00 2001 From: Boris de Jonge Date: Tue, 3 Feb 2026 11:31:22 +0100 Subject: [PATCH] fix: handle SampleWindow serialization (#542) --- macros/upload_individual_datasets/upload_invocations.sql | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/macros/upload_individual_datasets/upload_invocations.sql b/macros/upload_individual_datasets/upload_invocations.sql index 2391e1e0..3e97fd26 100644 --- a/macros/upload_individual_datasets/upload_invocations.sql +++ b/macros/upload_individual_datasets/upload_invocations.sql @@ -16,6 +16,14 @@ {%- do invocation_args_dict.update({"warn_error_options": warn_error_options}) %} {% endif %} {% endif %} + {# sample - returns a SampleWindow python object that's not JSON serializable #} + {% if "sample" in invocation_args_dict %} + {% if invocation_args_dict.sample is not string and invocation_args_dict.sample is not none %} + {% set sample_window = invocation_args_dict.sample %} + {% set sample_dict = {"start": sample_window.start | string, "end": sample_window.end | string} %} + {%- do invocation_args_dict.update({"sample": sample_dict}) %} + {% endif %} + {% endif %} {% set converted_invocation_args_dict = dbt_artifacts.safe_copy_mapping(invocation_args_dict) %} {{ return(adapter.dispatch("get_invocations_dml_sql", "dbt_artifacts")(converted_invocation_args_dict)) }}