Skip to content

Commit

Permalink
Satisfy mypy
Browse files Browse the repository at this point in the history
  • Loading branch information
dandavison committed Aug 26, 2024
1 parent 70b466d commit 271eefb
Showing 1 changed file with 9 additions and 2 deletions.
11 changes: 9 additions & 2 deletions tests/test_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import os
import uuid
from datetime import datetime, timedelta, timezone
from typing import Any, List, Optional, Tuple, cast
from typing import Any, List, Mapping, Optional, Tuple, cast
from unittest import mock

import google.protobuf.any_pb2
Expand Down Expand Up @@ -311,7 +311,14 @@ class start_workflow_execution(
def __init__(self) -> None:
pass

async def __call__(self, *args, **kwargs) -> google.protobuf.message.Message:
async def __call__(
self,
req: temporalio.api.workflowservice.v1.StartWorkflowExecutionRequest,
*,
retry: bool = False,
metadata: Mapping[str, str] = {},
timeout: Optional[timedelta] = None,
) -> temporalio.api.workflowservice.v1.StartWorkflowExecutionResponse:
raise self.already_exists_err

workflow_service: temporalio.service.WorkflowService = (
Expand Down

0 comments on commit 271eefb

Please sign in to comment.