fixed undeclared types

Author: billrich2001

batch_sliding_window/batch_workflow.py

@@ -1,5 +1,6 @@
 import asyncio
 from dataclasses import dataclass
+from datetime import timedelta
 from typing import List
 
 from temporalio import workflow
@@ -37,9 +38,9 @@ class ProcessBatchWorkflow:
     @workflow.run
     async def run(self, input: ProcessBatchWorkflowInput) -> int:
         # Get total record count
-        record_count = await workflow.execute_activity(
+        record_count: int = await workflow.execute_activity_method(
             RecordLoader.get_record_count,
-            start_to_close_timeout=workflow.timedelta(seconds=5),
+            start_to_close_timeout=timedelta(seconds=5),
         )
 
         if input.sliding_window_size < input.partitions:

batch_sliding_window/sliding_window_workflow.py

@@ -1,12 +1,14 @@
 import asyncio
 from dataclasses import dataclass
+from datetime import timedelta
 from typing import Dict, List, Optional, Set
 
 from temporalio import workflow
 from temporalio.common import WorkflowIDReusePolicy
 
 from batch_sliding_window.record_loader_activity import (
     GetRecordsInput,
+    GetRecordsOutput,
     RecordLoader,
     SingleRecord,
 )
@@ -86,10 +88,12 @@ async def _execute(self, input: SlidingWindowWorkflowInput) -> int:
                 offset=self.offset,
                 max_offset=input.maximum_offset,
             )
-            get_records_output = await workflow.execute_activity(
-                RecordLoader.get_records,
-                get_records_input,
-                start_to_close_timeout=workflow.timedelta(seconds=5),
+            get_records_output: GetRecordsOutput = (
+                await workflow.execute_activity_method(
+                    RecordLoader.get_records,
+                    get_records_input,
+                    start_to_close_timeout=timedelta(seconds=5),
+                )
             )
             records = get_records_output.records