Changed upsert_institution to return a session.get()

cfpb · Jan 11, 2024 · a989931 · a989931
1 parent 22dc087
commit a989931
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/entities/repos/institutions_repo.py b/src/entities/repos/institutions_repo.py
@@ -83,9 +83,7 @@ async def upsert_institution(session: AsyncSession, fi: FinancialInstitutionDto)
             del fi_data["sbl_institution_type_ids"]
 
         db_fi = await session.merge(FinancialInstitutionDao(**fi_data))
-        await session.flush([db_fi])
-        await session.refresh(db_fi)
-        return db_fi
+        return await session.get(FinancialInstitutionDao, db_fi.lei)
 
 
 async def add_domains(