.

web/main.py

@@ -8,7 +8,7 @@ from sqlalchemy.orm import Session
 import os
 from dotenv import load_dotenv
 
-from rota.shifts import SingleShift, ShiftConstraint
+from rota_generator.shifts import SingleShift, ShiftConstraint
 
 from models import SingleShift as ModelSingleShift
 #from models import ShiftConstraint as ModelShiftConstraint

web/schemas.py

@@ -1,5 +1,5 @@
-from rota.shifts import SingleShift as SingleShiftBase, ShiftConstraint as ShiftConstraintBase
-from rota.workers import Worker as WorkerBase
+from rota_generator.shifts import SingleShift as SingleShiftBase, ShiftConstraint as ShiftConstraintBase
+from rota_generator.workers import Worker as WorkerBase
 
 
 class SingleShift(SingleShiftBase):