Bugfix/SK-1431 | Change back from control.models to control.model (#818)

scaleoutsystems · Feb 20, 2025 · 24d392c · 24d392c
1 parent c11b238
commit 24d392c
Showing 1 changed file with 21 additions and 11 deletions.
diff --git a/fedn/network/storage/dbconnection.py b/fedn/network/storage/dbconnection.py
@@ -11,17 +11,27 @@
 from sqlalchemy.orm import sessionmaker
 
 from fedn.common.config import get_network_config, get_statestore_config
-from fedn.network.storage.statestore.stores.analytic_store import AnalyticStore, MongoDBAnalyticStore
-from fedn.network.storage.statestore.stores.client_store import ClientStore, MongoDBClientStore, SQLClientStore
-from fedn.network.storage.statestore.stores.combiner_store import CombinerStore, MongoDBCombinerStore, SQLCombinerStore
-from fedn.network.storage.statestore.stores.model_store import ModelStore, MongoDBModelStore, SQLModelStore
-from fedn.network.storage.statestore.stores.package_store import MongoDBPackageStore, PackageStore, SQLPackageStore
-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.session_store import MongoDBSessionStore, SessionStore, SQLSessionStore
-from fedn.network.storage.statestore.stores.status_store import MongoDBStatusStore, SQLStatusStore, StatusStore
+from fedn.network.storage.statestore.stores.analytic_store import (
+    AnalyticStore, MongoDBAnalyticStore)
+from fedn.network.storage.statestore.stores.client_store import (
+    ClientStore, MongoDBClientStore, SQLClientStore)
+from fedn.network.storage.statestore.stores.combiner_store import (
+    CombinerStore, MongoDBCombinerStore, SQLCombinerStore)
+from fedn.network.storage.statestore.stores.model_store import (
+    ModelStore, MongoDBModelStore, SQLModelStore)
+from fedn.network.storage.statestore.stores.package_store import (
+    MongoDBPackageStore, PackageStore, SQLPackageStore)
+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.session_store import (
+    MongoDBSessionStore, SessionStore, SQLSessionStore)
+from fedn.network.storage.statestore.stores.status_store import (
+    MongoDBStatusStore, SQLStatusStore, StatusStore)
 from fedn.network.storage.statestore.stores.store import MyAbstractBase
-from fedn.network.storage.statestore.stores.validation_store import MongoDBValidationStore, SQLValidationStore, ValidationStore
+from fedn.network.storage.statestore.stores.validation_store import (
+    MongoDBValidationStore, SQLValidationStore, ValidationStore)
 
 
 class StoreContainer:
@@ -89,7 +99,7 @@ def _init_connection(self) -> None:
             prediction_store = MongoDBPredictionStore(mdb, "control.predictions")
             round_store = MongoDBRoundStore(mdb, "control.rounds")
             package_store = MongoDBPackageStore(mdb, "control.packages")
-            model_store = MongoDBModelStore(mdb, "control.models")
+            model_store = MongoDBModelStore(mdb, "control.model")
             session_store = MongoDBSessionStore(mdb, "control.sessions")
             analytic_store = MongoDBAnalyticStore(mdb, "control.analytics")