Update start time calculation for SageMaker Orchestrator

zenml-io · Jan 9, 2025 · b995970 · b995970
1 parent 48f93ff
commit b995970
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py b/src/zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py
@@ -516,7 +516,8 @@ def prepare_or_run_pipeline(
                     enabled=True,
                 )
                 next_execution = (
-                    deployment.schedule.start_time or datetime.utcnow()
+                    deployment.schedule.start_time
+                    or datetime.now(timezone.utc)
                 ) + deployment.schedule.interval_second
             else:
                 # One-time schedule