Add tox

[pti/o2.git] / o2ims / service / unit_of_work.py

diff --git a/src/o2ims/service/unit_of_work.py b/o2ims/service/unit_of_work.py
similarity index 94%
rename from src/o2ims/service/unit_of_work.py
rename to o2ims/service/unit_of_work.py
index 4714729..40e0f76 100644 (file)
--- a/src/o2ims/service/unit_of_work.py
+++ b/o2ims/service/unit_of_work.py
@@ -19,7 +19,6 @@ from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
 from sqlalchemy.orm.session import Session
 
-
 from o2ims import config
 from o2ims.adapter import ocloud_repository
 
@@ -27,7 +26,7 @@ from o2ims.adapter import ocloud_repository
 class AbstractUnitOfWork(abc.ABC):
     oclouds: ocloud_repository.OcloudRepository
 
-    def __enter__(self) -> AbstractUnitOfWork:
+    def __enter__(self):
         return self
 
     def __exit__(self, *args):
@@ -64,7 +63,8 @@ class SqlAlchemyUnitOfWork(AbstractUnitOfWork):
 
     def __enter__(self):
         self.session = self.session_factory()  # type: Session
-        self.oclouds = ocloud_repository.OcloudSqlAlchemyRepository(self.session)
+        self.oclouds = ocloud_repository\
+            .OcloudSqlAlchemyRepository(self.session)
         return super().__enter__()
 
     def __exit__(self, *args):