Update Pendulum imports

The-Academic-Observatory · Mar 17, 2024 · c2f1da2 · c2f1da2
1 parent 6979926
commit c2f1da2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/observatory_platform/dataset_api.py b/observatory_platform/dataset_api.py
@@ -314,12 +314,12 @@ def build_schedule(sched_start_date: pendulum.DateTime, sched_end_date: pendulum
 
     schedule = []
 
-    for start_date in pendulum.Period(start=sched_start_date, end=sched_end_date).range("months"):
+    for start_date in pendulum.Interval(start=sched_start_date, end=sched_end_date).range("months"):
         if start_date >= sched_end_date:
             break
         end_date = start_date.add(months=1).subtract(days=1).end_of("day")
         end_date = min(sched_end_date, end_date)
-        schedule.append(pendulum.Period(start_date.date(), end_date.date()))
+        schedule.append(pendulum.Interval(start_date.date(), end_date.date()))
 
     return schedule
 

diff --git a/observatory_platform/tests/test_dataset_api.py b/observatory_platform/tests/test_dataset_api.py
@@ -170,20 +170,20 @@ def test_build_schedule(self):
         start_date = pendulum.datetime(2021, 1, 1)
         end_date = pendulum.datetime(2021, 2, 1)
         schedule = build_schedule(start_date, end_date)
-        self.assertEqual([pendulum.Period(pendulum.date(2021, 1, 1), pendulum.date(2021, 1, 31))], schedule)
+        self.assertEqual([pendulum.Interval(pendulum.date(2021, 1, 1), pendulum.date(2021, 1, 31))], schedule)
 
         start_date = pendulum.datetime(2021, 1, 1)
         end_date = pendulum.datetime(2021, 3, 1)
         schedule = build_schedule(start_date, end_date)
         self.assertEqual(
             [
-                pendulum.Period(pendulum.date(2021, 1, 1), pendulum.date(2021, 1, 31)),
-                pendulum.Period(pendulum.date(2021, 2, 1), pendulum.date(2021, 2, 28)),
+                pendulum.Interval(pendulum.date(2021, 1, 1), pendulum.date(2021, 1, 31)),
+                pendulum.Interval(pendulum.date(2021, 2, 1), pendulum.date(2021, 2, 28)),
             ],
             schedule,
         )
 
         start_date = pendulum.datetime(2021, 1, 7)
         end_date = pendulum.datetime(2021, 2, 7)
         schedule = build_schedule(start_date, end_date)
-        self.assertEqual([pendulum.Period(pendulum.date(2021, 1, 7), pendulum.date(2021, 2, 6))], schedule)
+        self.assertEqual([pendulum.Interval(pendulum.date(2021, 1, 7), pendulum.date(2021, 2, 6))], schedule)