fixing sagemaker (#3380)

zenml-io · Feb 26, 2025 · f16e28a · f16e28a
1 parent 3d484c2
commit f16e28a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py b/src/zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py
@@ -472,7 +472,10 @@ def prepare_or_run_pipeline(
 
             # Convert environment to a dict of strings
             environment = {
-                key: str(value) for key, value in environment.items()
+                key: str(value)
+                if not isinstance(value, ExecutionVariables)
+                else value
+                for key, value in environment.items()
             }
 
             if use_training_step: