run table creation script from combiner to fix timing issue

scaleoutsystems · Jan 9, 2025 · 669c154 · 669c154
1 parent b8f89b7
commit 669c154
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/fedn/network/combiner/shared.py b/fedn/network/combiner/shared.py
@@ -9,6 +9,7 @@
 from fedn.network.storage.statestore.stores.prediction_store import MongoDBPredictionStore, PredictionStore, SQLPredictionStore
 from fedn.network.storage.statestore.stores.round_store import MongoDBRoundStore, RoundStore, SQLRoundStore
 from fedn.network.storage.statestore.stores.status_store import MongoDBStatusStore, SQLStatusStore, StatusStore
+from fedn.network.storage.statestore.stores.store import MyAbstractBase, engine
 from fedn.network.storage.statestore.stores.validation_store import MongoDBValidationStore, SQLValidationStore, ValidationStore
 
 statestore_config = get_statestore_config()
@@ -34,6 +35,8 @@
     prediction_store = MongoDBPredictionStore(mdb, "control.predictions")
     round_store = MongoDBRoundStore(mdb, "control.rounds")
 elif statestore_config["type"] in ["SQLite", "PostgreSQL"]:
+    MyAbstractBase.metadata.create_all(engine, checkfirst=True)
+
     client_store = SQLClientStore()
     validation_store = SQLValidationStore()
     combiner_store = SQLCombinerStore()