diff --git a/src/saturn_engine/worker/worker_manager.py b/src/saturn_engine/worker/worker_manager.py index 92b0a57b..fb8a4abb 100644 --- a/src/saturn_engine/worker/worker_manager.py +++ b/src/saturn_engine/worker/worker_manager.py @@ -30,7 +30,7 @@ def __init__( self.context = WorkerManagerContext(config=config.c.worker_manager) with self.sessionmaker() as session: - self.context.reset_static_definition(session=session) + self.context.load_static_definition(session=session) async def init_db(self) -> None: return await asyncio.get_event_loop().run_in_executor( diff --git a/src/saturn_engine/worker_manager/context.py b/src/saturn_engine/worker_manager/context.py index 5b8726e0..07710554 100644 --- a/src/saturn_engine/worker_manager/context.py +++ b/src/saturn_engine/worker_manager/context.py @@ -18,9 +18,8 @@ def __init__(self, config: WorkerManagerConfig) -> None: @property def static_definitions(self) -> StaticDefinitions: - assert ( - self._static_definitions - ), "Static definitions need to be set before usage" + if not self._static_definitions: + raise ValueError("Static definitions need to be set before usage") return self._static_definitions def load_static_definition(self, session: AnySession) -> None: diff --git a/src/saturn_engine/worker_manager/server.py b/src/saturn_engine/worker_manager/server.py index 83f7b6a8..8e574883 100644 --- a/src/saturn_engine/worker_manager/server.py +++ b/src/saturn_engine/worker_manager/server.py @@ -56,7 +56,7 @@ def init_all(app: Optional[SaturnApp] = None) -> None: with app.app_context(): create_all() with session_scope() as session: - current_app.saturn.reset_static_definition(session=session) + current_app.saturn.load_static_definition(session=session) sync_jobs( static_definitions=current_app.saturn.static_definitions, session=session,