Fix changes from workflow info being optional

Author: tconley1428

custom_metric/worker.py

@@ -39,7 +39,7 @@ async def execute_activity(self, input: ExecuteActivityInput):
             unit="duration",
         )
         histogram.record(
-            schedule_to_start, {"workflow_type": activity.info().workflow_type}
+            schedule_to_start, {"workflow_type": activity.info().workflow_type or ""}
         )
         return await self.next.execute_activity(input)
 

polling/test_service.py

@@ -4,7 +4,7 @@
 from temporalio import activity
 from temporalio.exceptions import ApplicationError, ApplicationErrorCategory
 
-attempts = Counter[str]()
+attempts = Counter[str | None]()
 ERROR_ATTEMPTS = 5
 
 

tests/resource_pool/workflow_test.py

@@ -28,7 +28,7 @@ async def test_resource_pool_workflow(client: Client):
     # Mock out the activity to count executions
     @activity.defn(name="use_resource")
     async def use_resource_mock(input: UseResourceActivityInput) -> None:
-        workflow_id = activity.info().workflow_id
+        workflow_id = activity.info().workflow_id or ""
         resource_usage[input.resource].append((workflow_id, "start"))
         # We need a small sleep here to bait out races
         await asyncio.sleep(0.05)