Add WorkflowThread where applicable

littlehorse-enterprises · Nov 27, 2024 · 1821425 · 1821425
1 parent 39c45f8
commit 1821425
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdk-python/tests/test_workflow.py b/sdk-python/tests/test_workflow.py
@@ -2171,7 +2171,7 @@ def wf_func(thread: WorkflowThread) -> None:
 
     def test_reassign_to_user_var(self):
         def wf_func(thread: WorkflowThread) -> None:
-            user_var = WfRunVariable("my-var", VariableType.STR, None)
+            user_var = WfRunVariable("my-var", VariableType.STR, thread)
             uto = thread.assign_user_task(
                 "my-user-task",
                 user_id="asdf",
@@ -2197,7 +2197,7 @@ def wf_func(thread: WorkflowThread) -> None:
 
     def test_reassign_to_group(self):
         def wf_func(thread: WorkflowThread) -> None:
-            user_var = WfRunVariable("my-var", VariableType.STR, None)
+            user_var = WfRunVariable("my-var", VariableType.STR, thread)
             uto = thread.assign_user_task(
                 "my-user-task",
                 user_id="asdf",