diff --git a/backend/categorisation/processor.py b/backend/categorisation/processor.py index 9c1bb8f0..ee42efcd 100644 --- a/backend/categorisation/processor.py +++ b/backend/categorisation/processor.py @@ -1,12 +1,9 @@ from collections import defaultdict -from typing import List, cast - -from sqlalchemy import Tuple +from typing import List, Tuple, cast from backend.app.db.functions.recommendations_functions import ( get_plans_by_portfolio_id, get_scenario, - set_plan_default, update_plan, ) from backend.app.db.models.recommendations import PlanModel, ScenarioModel @@ -65,7 +62,7 @@ def _update_default_flags(plans: List[Plan], cheapest_plan: Plan) -> None: plan.set_default(plan.id == cheapest_plan.id) plan_model, scenario_model = cast( - tuple[PlanModel, ScenarioModel], + Tuple[PlanModel, ScenarioModel], plan.to_sqlalchemy(), )