fix comment

Signed-off-by: Tim Li <[email protected]>

Author: timl3136

cadence/_internal/workflow/decision_events_iterator.py

@@ -157,10 +157,10 @@ async def next_decision_events(self) -> DecisionEvents:
         decision_events.events.append(decision_task_started)
 
         # Update replay time if available
-        if hasattr(decision_task_started, 'event_time') and decision_task_started.event_time:
-            self._replay_current_time_milliseconds = getattr(
-                decision_task_started.event_time, 'seconds', 0
-            ) * 1000
+        if decision_task_started.event_time:
+            self._replay_current_time_milliseconds = (
+                decision_task_started.event_time.seconds * 1000
+            )
             decision_events.replay_current_time_milliseconds = self._replay_current_time_milliseconds
 
         # Process subsequent events until we find the corresponding DecisionTask completion

cadence/worker/_decision_task_handler.py

@@ -112,8 +112,7 @@ async def _handle_task_implementation(self, task: PollForDecisionTaskResponse) -
         decision_result = await workflow_engine.process_decision(task)
 
         # Clean up completed workflows from cache to prevent memory leaks
-        # Use getattr with default False to handle mocked engines in tests
-        if getattr(workflow_engine, '_is_workflow_complete', False):
+        if workflow_engine._is_workflow_complete:
             with self._cache_lock:
                 self._workflow_engines.pop(cache_key, None)
                 logger.debug(