fix(sc): preserve cancellation during replay buffer rollback

Signed-off-by: ruit <ruit@nvidia.com>

Author: RayenTian

nemo_rl/algorithms/async_utils/replay_buffer.py

@@ -755,6 +755,8 @@ async def commit(
                     partition_id=self._partition_id,
                 )
             except BaseException as rollback_error:
+                if isinstance(commit_error, asyncio.CancelledError):
+                    raise commit_error from rollback_error
                 raise BaseExceptionGroup(
                     f"commit and rollback both failed for group_id={group_id!r}",
                     [commit_error, rollback_error],