Remove none values from config template

schustmi · schustmi · commit bdaf3858bcea · 2025-04-23T13:03:26.000+02:00
diff --git a/src/zenml/zen_stores/template_utils.py b/src/zenml/zen_stores/template_utils.py
@@ -99,6 +99,7 @@ def generate_config_template(
         name: step.config.model_dump(
             include=set(StepConfigurationUpdate.model_fields),
             exclude={"name", "outputs"},
+            exclude_none=True,
         )
         for name, step in deployment_model.step_configurations.items()
     }
@@ -109,6 +110,7 @@ def generate_config_template(
     pipeline_config = deployment_model.pipeline_configuration.model_dump(
         include=set(PipelineRunConfiguration.model_fields),
         exclude={"schedule", "build", "parameters"},
+        exclude_none=True,
     )
 
     pipeline_config["settings"].pop("docker", None)

Original file line number	Diff line number	Diff line change
`@@ -99,6 +99,7 @@ def generate_config_template(`
`99`	`99`	`name: step.config.model_dump(`
`100`	`100`	`include=set(StepConfigurationUpdate.model_fields),`
`101`	`101`	`exclude={"name", "outputs"},`
	`102`	`+ exclude_none=True,`
`102`	`103`	`)`
`103`	`104`	`for name, step in deployment_model.step_configurations.items()`
`104`	`105`	`}`
`@@ -109,6 +110,7 @@ def generate_config_template(`
`109`	`110`	`pipeline_config = deployment_model.pipeline_configuration.model_dump(`
`110`	`111`	`include=set(PipelineRunConfiguration.model_fields),`
`111`	`112`	`exclude={"schedule", "build", "parameters"},`
	`113`	`+ exclude_none=True,`
`112`	`114`	`)`
`113`	`115`
`114`	`116`	`pipeline_config["settings"].pop("docker", None)`