From 2c55ab4ad80a0cd1f55000a2114c981cbc9c5fd7 Mon Sep 17 00:00:00 2001
From: admin user <habemco_admin_user@centurion.com>
Date: Fri, 23 May 2025 02:03:18 +0000
Subject: [PATCH] Update templates/workflow_template.py.j2

---
 templates/workflow_template.py.j2 | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/templates/workflow_template.py.j2 b/templates/workflow_template.py.j2
index 25fb368..fbfeee7 100644
--- a/templates/workflow_template.py.j2
+++ b/templates/workflow_template.py.j2
@@ -6,7 +6,7 @@ import json
 import datetime
 import re
 import jmespath
-from temporalio import workflow
+from temporalio.workflow import sleep
 
 # Configure logging
 logging.basicConfig(level=logging.INFO,
@@ -17,7 +17,7 @@ logger = logging.getLogger(__name__)
 class {{ workflow_class_name }}:
     @temporalio.workflow.run
     async def run(self, root_inputs: Dict[str, Any]) -> Dict[str, Any]:
-        await workflow.sleep(0)
+        await sleep(0)
         workflow_info = temporalio.workflow.info()
         workflow_output: Dict[str, Any] = {
             "workflow_id": workflow_info.workflow_id,