fix after review

Author: Вадим Козыревский

src/cqrs/saga/saga.py

@@ -168,12 +168,8 @@ async def __aiter__(
             run_cm = None
 
         if run_cm is not None:
-            try:
-                async with run_cm as run:
-                    async for step_result in self._execute(run):
-                        yield step_result
-            except NotImplementedError:
-                async for step_result in self._execute(None):
+            async with run_cm as run:
+                async for step_result in self._execute(run):
                     yield step_result
         else:
             async for step_result in self._execute(None):

tests/integration/test_saga_mediator_sqlalchemy_postgres.py

@@ -11,8 +11,8 @@
 from cqrs.saga.storage.enums import SagaStatus
 from cqrs.saga.storage.sqlalchemy import SqlAlchemySagaStorage
 
+from tests.integration.conftest import _TestContainer
 from tests.integration.test_saga_mediator_memory import (
-    _TestContainer,
     FailingOrderSaga,
     InventoryReservedEventHandler,
     OrderContext,