diff --git a/timetable-solver-service/app/runner.py b/timetable-solver-service/app/runner.py index 70e0df4..cd26f3f 100644 --- a/timetable-solver-service/app/runner.py +++ b/timetable-solver-service/app/runner.py @@ -20,6 +20,7 @@ from concurrent.futures import ThreadPoolExecutor from timefold.solver import SolverFactory from timefold.solver.config import ( SolverConfig, + ScoreDirectorFactoryConfig, TerminationConfig, Duration, ) @@ -38,7 +39,9 @@ _solver_factory = SolverFactory.create( SolverConfig( solution_class=Timetable, entity_class_list=[Lesson], - score_director_factory_config={"constraint_provider_function": define_constraints}, + score_director_factory_config=ScoreDirectorFactoryConfig( + constraint_provider_function=define_constraints, + ), termination_config=TerminationConfig( spent_limit=Duration(seconds=settings.solver_seconds_limit), ),