Fix lint issues

legal-api/src/legal_api/models/amalgamation.py

@@ -19,7 +19,7 @@
 
 from enum import auto
 
-from sql_versioning import history_cls, Versioned
+from sql_versioning import Versioned, history_cls
 
 from ..utils.enum import BaseEnum
 from .db import db
@@ -95,7 +95,7 @@ def get_amalgamation_revision_obj(cls, filing, legal_entity_id):
     def get_amalgamation_revision_json(cls, filing, legal_entity_id):
         """Get amalgamation json for the given transaction id."""
         from legal_api.services import business_service
-        
+
         amalgamation = Amalgamation.get_amalgamation_revision_obj(filing, legal_entity_id)
 
         business = business_service.fetch_business_by_filing(filing)

legal-api/src/legal_api/models/legal_entity.py

@@ -638,9 +638,7 @@ def get_amalgamated_into(self) -> dict:
             and (state_filing := Filing.find_by_id(self.state_filing_id))
             and state_filing.is_amalgamation_application
         ):
-            return Amalgamation.get_amalgamation_revision_json(
-                state_filing, state_filing.legal_entity_id
-            )
+            return Amalgamation.get_amalgamation_revision_json(state_filing, state_filing.legal_entity_id)
 
         return None
 

legal-api/tests/unit/models/test_legal_entity.py

@@ -547,13 +547,13 @@ def test_amalgamated_into_business_json(session, test_name, existing_business_st
         amalgamation.amalgamation_type = "regular"
         amalgamation.amalgamation_date = datetime.now()
         amalgamation.court_approval = True
+        amalgamation.save()
 
         amalgamating_business = AmalgamatingBusiness()
         amalgamating_business.role = "amalgamating"
         amalgamating_business.legal_entity_id = existing_business.id
         amalgamating_business.change_filing_id = filing.id
         amalgamation.amalgamating_businesses.append(amalgamating_business)
-        amalgamation.save()
 
         business.amalgamation.append(amalgamation)
         db.session.add(business)