diff --git a/temporalio/testing/_timeskipping.py b/temporalio/testing/_timeskipping.py index ab553e6c89..ab5a091d3a 100644 --- a/temporalio/testing/_timeskipping.py +++ b/temporalio/testing/_timeskipping.py @@ -186,6 +186,7 @@ async def fast_forward( id=fast_forward_id, duration=duration ), disable_propagation=self._config.disable_propagation, + max_session_skip_count=self._config.max_session_skip_count, ), ) return await self._poll_fast_forward_completion(handle, fast_forward_id) @@ -206,6 +207,7 @@ async def _wait_for_unbounded_fast_forward_completion( TimeSkippingConfig( enabled=True, disable_propagation=self._config.disable_propagation, + max_session_skip_count=self._config.max_session_skip_count, ), ) async for event in handle.fetch_history_events(wait_new_event=True):