diff --git a/app/helpers/solver_helper.py b/app/helpers/solver_helper.py index e754a42..1b1c613 100644 --- a/app/helpers/solver_helper.py +++ b/app/helpers/solver_helper.py @@ -12,7 +12,7 @@ from models.item import Item from lib.errors.item_generation_error import ItemGenerationError def build_constraints(solver_run: SolverRun, problem: LpProblem, - items: list[Item], bundles: list[Bundle], selected_items, selected_bundles) -> LpProblem: + items: list[Item], bundles: list[Bundle], selected_items: list[Item], selected_bundles: list[Bundle]) -> LpProblem: logging.info('Creating Constraints...') try: diff --git a/app/models/solver_run.py b/app/models/solver_run.py index a1a729b..ad54659 100644 --- a/app/models/solver_run.py +++ b/app/models/solver_run.py @@ -113,7 +113,7 @@ class SolverRun(BaseModel): else: return self.items - def select_items_by_percent(self, percent) -> list[Item]: + def select_items_by_percent(self, percent: int) -> list[Item]: items = self.unbundled_items() total_items = len(items) selected_items_amount = round(total_items - (total_items * @@ -121,7 +121,7 @@ class SolverRun(BaseModel): return random.sample(items, selected_items_amount) - def select_bundles_by_percent(self, percent) -> list[Bundle]: + def select_bundles_by_percent(self, percent: int) -> list[Bundle]: total_bundles = len(self.bundles) selected_bundles_amount = round(total_bundles - (total_bundles * (percent / 100)))