OR_tool shift scheduling: shift_scheduling_sat.py More that 3 shifts adjustment

[jbetkowska]

Dears,

I am new in this area, I tried to adjust code for shift scheduling by myself but seems that I stuck.
In my company we have 5 shifts: M, A, N from Monday to Sunday and W and Z in Saturday and Sunday.
I wanted to change this as follows:

1. Change the line of code with shifts definition: shifts = ["O", "M", "A", "N", "W", "Z"]
2. Change demand for work shifts in a way, that on Sat and Sun we have 0 demand for M, A, N, for other days we have 0 demand for W and Z (to solve the issue that we work in 8hrs shifts during the working days and in 12hrs shift during the weekends)
3. Change # Penalty for exceeding the cover constraint per shift type to have penalties for all 5 shifts excess_cover_penalties = (2, 2, 5, 2, 26)

However I got an error: IndexError: list index out of range
And I don't know how to adjust this code to handle this.
In attachment my file with other changes I've already made to fit my company's needs.
rota_scheduler.txt

[lperron]

6 shifts ?

[jbetkowska]

Technically yes, but in the original code shift "O" was omitted as it was off day for worker. So in original code we had shifts = ["O", "M", "A", "N"] and constraints were like for 3, I added 2 shifts and changed contraints for 5.

[lperron]

Which line throws the exception?