diff --git a/src/database/database_dummy.py b/src/database/database_dummy.py index b361fac..6635c76 100644 --- a/src/database/database_dummy.py +++ b/src/database/database_dummy.py @@ -18,10 +18,7 @@ def get_lead_by_id(self, id_: int) -> Lead: return LeadParser.parse_lead_from_dict(self.data[id_]) def get_all_leads(self) -> List[Lead]: - leads = [] - for entry in self.data.values(): - leads.append(LeadParser.parse_lead_from_dict(entry)) - return leads + return [LeadParser.parse_lead_from_dict(entry) for entry in self.data.values()] def update_lead(self, lead: Lead): print(f"Updating database entry for lead#{lead.lead_id}")