From c26fb4d7fa651e69a490907842480bd121d8bf8c Mon Sep 17 00:00:00 2001
From: Olivier Michaud <olivier.michaud@flare.systems>
Date: Wed, 16 Oct 2024 15:56:35 -0400
Subject: [PATCH] Fix broken function call

---
 src/saturn_engine/worker/worker_manager.py  | 2 +-
 src/saturn_engine/worker_manager/context.py | 5 ++---
 src/saturn_engine/worker_manager/server.py  | 2 +-
 3 files changed, 4 insertions(+), 5 deletions(-)

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,