Minor fixes

README.md

@@ -1230,7 +1230,12 @@ poe test
 
 This runs against [Temporalite](https://github.com/temporalio/temporalite). To run against the time-skipping test
 server, pass `--workflow-environment time-skipping`. To run against the `default` namespace of an already-running
-server, pass the `host:port` to `--workflow-environment`.
+server, pass the `host:port` to `--workflow-environment`. Can also use regular pytest arguments. For example, here's how
+to run a single test with debug logs on the console:
+
+```bash
+poe test -s --log-cli-level=DEBUG -k test_sync_activity_thread_cancel_caught
+```
 
 #### Proto Generation and Testing
 

pyproject.toml

@@ -35,11 +35,11 @@ types-protobuf = "~3.20"
 typing-extensions = "^4.2.0"
 
 [tool.poetry.dev-dependencies]
-black = "^22.3.0"
+black = "^23.1.0"
 cibuildwheel = "^2.11.0"
 grpcio-tools = "^1.48.0"
-isort = "^5.11.3"
-mypy = "^0.971"
+isort = "^5.11.5"
+mypy = "^1.0.0"
 mypy-protobuf = "^3.3.0"
 protoc-wheel-0 = "^21.1"
 psutil = "^5.9.3"
@@ -76,7 +76,7 @@ lint = [
 # TODO(cretz): Why does pydocstyle complain about @overload missing docs after
 # https://github.com/PyCQA/pydocstyle/pull/511?
 lint-docs = "pydocstyle --ignore-decorators=overload"
-lint-types = "mypy --namespace-packages ."
+lint-types = "mypy --namespace-packages --check-untyped-defs ."
 run-bench = "python scripts/run_bench.py"
 test = "pytest"
 

scripts/run_bench.py

@@ -14,6 +14,8 @@
 from temporalio.worker import UnsandboxedWorkflowRunner, Worker
 from temporalio.worker.workflow_sandbox import SandboxedWorkflowRunner
 
+assert sys.version_info >= (3, 9)
+
 
 @workflow.defn
 class BenchWorkflow:

temporalio/api/command/v1/message_pb2.pyi

@@ -1148,24 +1148,27 @@ class Command(google.protobuf.message.Message):
     ) -> None: ...
     def WhichOneof(
         self, oneof_group: typing_extensions.Literal["attributes", b"attributes"]
-    ) -> typing_extensions.Literal[
-        "schedule_activity_task_command_attributes",
-        "start_timer_command_attributes",
-        "complete_workflow_execution_command_attributes",
-        "fail_workflow_execution_command_attributes",
-        "request_cancel_activity_task_command_attributes",
-        "cancel_timer_command_attributes",
-        "cancel_workflow_execution_command_attributes",
-        "request_cancel_external_workflow_execution_command_attributes",
-        "record_marker_command_attributes",
-        "continue_as_new_workflow_execution_command_attributes",
-        "start_child_workflow_execution_command_attributes",
-        "signal_external_workflow_execution_command_attributes",
-        "upsert_workflow_search_attributes_command_attributes",
-        "accept_workflow_update_command_attributes",
-        "complete_workflow_update_command_attributes",
-        "modify_workflow_properties_command_attributes",
-        "reject_workflow_update_command_attributes",
-    ] | None: ...
+    ) -> (
+        typing_extensions.Literal[
+            "schedule_activity_task_command_attributes",
+            "start_timer_command_attributes",
+            "complete_workflow_execution_command_attributes",
+            "fail_workflow_execution_command_attributes",
+            "request_cancel_activity_task_command_attributes",
+            "cancel_timer_command_attributes",
+            "cancel_workflow_execution_command_attributes",
+            "request_cancel_external_workflow_execution_command_attributes",
+            "record_marker_command_attributes",
+            "continue_as_new_workflow_execution_command_attributes",
+            "start_child_workflow_execution_command_attributes",
+            "signal_external_workflow_execution_command_attributes",
+            "upsert_workflow_search_attributes_command_attributes",
+            "accept_workflow_update_command_attributes",
+            "complete_workflow_update_command_attributes",
+            "modify_workflow_properties_command_attributes",
+            "reject_workflow_update_command_attributes",
+        ]
+        | None
+    ): ...
 
 global___Command = Command

temporalio/api/failure/v1/message_pb2.pyi

@@ -423,15 +423,18 @@ class Failure(google.protobuf.message.Message):
     ) -> None: ...
     def WhichOneof(
         self, oneof_group: typing_extensions.Literal["failure_info", b"failure_info"]
-    ) -> typing_extensions.Literal[
-        "application_failure_info",
-        "timeout_failure_info",
-        "canceled_failure_info",
-        "terminated_failure_info",
-        "server_failure_info",
-        "reset_workflow_failure_info",
-        "activity_failure_info",
-        "child_workflow_execution_failure_info",
-    ] | None: ...
+    ) -> (
+        typing_extensions.Literal[
+            "application_failure_info",
+            "timeout_failure_info",
+            "canceled_failure_info",
+            "terminated_failure_info",
+            "server_failure_info",
+            "reset_workflow_failure_info",
+            "activity_failure_info",
+            "child_workflow_execution_failure_info",
+        ]
+        | None
+    ): ...
 
 global___Failure = Failure