Fixed linking to datasrc (#297)

* Fixed linking to datasrc * Removed logging comments
truefoundry · Aug 9, 2024 · 2ac8821 · 2ac8821
1 parent 3578d83
commit 2ac8821
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/backend/modules/metadata_store/prismastore.py b/backend/modules/metadata_store/prismastore.py
@@ -231,18 +231,18 @@ async def aassociate_data_source_with_collection(
             if existing_collection_associated_data_sources:
                 existing_collection_associated_data_sources[
                     data_src_to_associate.data_source_fqn
-                ] = data_src_to_associate
+                ] = data_src_to_associate.dict()
             else:
                 existing_collection_associated_data_sources = {
-                    data_src_to_associate.data_source_fqn: data_src_to_associate
+                    data_src_to_associate.data_source_fqn: data_src_to_associate.dict()
                 }
 
             associated_data_sources: Dict[str, Dict[str, Any]] = {}
             for (
                 data_source_fqn,
                 data_source,
             ) in existing_collection_associated_data_sources.items():
-                associated_data_sources[data_source_fqn] = data_source.dict()
+                associated_data_sources[data_source_fqn] = data_source
 
             updated_collection = await self.db.collection.update(
                 where={"name": collection_name},