.

test/test_general_balancing.py

@@ -1,8 +1,8 @@
 import datetime
 import pytest
 from pytest import approx
-from rota.shifts import PostShiftConstraint, PreShiftConstraint, RotaBuilder, SingleShift, days
-from rota.workers import Worker
+from rota_generator.shifts import PostShiftConstraint, PreShiftConstraint, RotaBuilder, SingleShift, days
+from rota_generator.workers import Worker
 
 def generate_basic_rota(weeks_to_rota=10):
     start_date = datetime.date(2022, 3, 7)