Skip to content

Commit 7536567

Browse files
committed
unpack params into function arguments
1 parent 5af20b6 commit 7536567

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

app.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
from __future__ import annotations
1515

1616
import multiprocess
17+
from dataclasses import asdict
1718
from typing import TYPE_CHECKING, Any
1819

1920
import diskcache
@@ -355,16 +356,16 @@ def run_optimization(
355356
solver_type = SolverType(solver)
356357

357358
if solver_type is SolverType.CQM:
358-
cqm = employee_scheduling.build_cqm(params)
359+
cqm = employee_scheduling.build_cqm(**asdict(params))
359360

360361
feasible_sampleset, errors = employee_scheduling.run_cqm(cqm)
361362
sample = feasible_sampleset.first.sample
362363

363364
sched = utils.build_schedule_from_sample(sample, employees)
364365

365366
elif solver_type is SolverType.NL:
366-
model, assignments = employee_scheduling.build_nl(params)
367-
errors = employee_scheduling.run_nl(model, assignments, params)
367+
model, assignments = employee_scheduling.build_nl(**asdict(params))
368+
errors = employee_scheduling.run_nl(model, assignments, **asdict(params))
368369
sched = utils.build_schedule_from_state(assignments.state(), employees, shifts)
369370

370371
else:

0 commit comments

Comments
 (0)