diff --git a/packages/syft/src/syft/store/mongo_client.py b/packages/syft/src/syft/store/mongo_client.py index d605ea86e15..c8af9b97c75 100644 --- a/packages/syft/src/syft/store/mongo_client.py +++ b/packages/syft/src/syft/store/mongo_client.py @@ -1,7 +1,8 @@ # relative from ..serde.serializable import serializable +from .document_store import StoreClientConfig @serializable(canonical_name="MongoStoreClientConfig", version=1) -class MongoStoreClientConfig: +class MongoStoreClientConfig(StoreClientConfig): pass diff --git a/packages/syft/src/syft/store/mongo_document_store.py b/packages/syft/src/syft/store/mongo_document_store.py index 80f9ba3a341..18eb99d89b4 100644 --- a/packages/syft/src/syft/store/mongo_document_store.py +++ b/packages/syft/src/syft/store/mongo_document_store.py @@ -14,7 +14,7 @@ from .mongo_client import MongoStoreClientConfig -class MongoBsonObject: +class MongoBsonObject(StorableObjectType): pass