|
5 | 5 |
|
6 | 6 | import pytest
|
7 | 7 |
|
8 |
| -from temporalio import activity, workflow |
| 8 | +from temporalio import activity, worker, workflow |
9 | 9 | from temporalio.client import Client, WorkflowUpdateFailedError
|
10 | 10 | from temporalio.exceptions import ApplicationError
|
11 | 11 | from temporalio.testing import WorkflowEnvironment
|
@@ -204,6 +204,8 @@ async def test_worker_interceptor(client: Client, env: WorkflowEnvironment):
|
204 | 204 | workflows=[InterceptedWorkflow],
|
205 | 205 | activities=[intercepted_activity],
|
206 | 206 | interceptors=[TracingWorkerInterceptor()],
|
| 207 | + # TODO: This test shouldn't need it, but frequently fails with strange import issues |
| 208 | + workflow_runner=worker.UnsandboxedWorkflowRunner(), |
207 | 209 | ):
|
208 | 210 | # Run workflow
|
209 | 211 | handle = await client.start_workflow(
|
@@ -317,6 +319,8 @@ async def test_workflow_instance_access_from_interceptor(client: Client):
|
317 | 319 | task_queue=task_queue,
|
318 | 320 | workflows=[WorkflowInstanceAccessWorkflow],
|
319 | 321 | interceptors=[WorkflowInstanceAccessInterceptor()],
|
| 322 | + # TODO: This test shouldn't need it, but frequently fails with strange import issues |
| 323 | + workflow_runner=worker.UnsandboxedWorkflowRunner(), |
320 | 324 | ):
|
321 | 325 | difference = await client.execute_workflow(
|
322 | 326 | WorkflowInstanceAccessWorkflow.run,
|
|
0 commit comments