rm pendulum use in schedules.py (#16943)

PrefectHQ · Feb 3, 2025 · 33dc16e · 33dc16e
1 parent 52c77b3
commit 33dc16e
Showing 1 changed file with 6 additions and 16 deletions.
diff --git a/src/prefect/client/schemas/schedules.py b/src/prefect/client/schemas/schedules.py
@@ -8,7 +8,6 @@
 import dateutil
 import dateutil.rrule
 import dateutil.tz
-import pendulum
 from pydantic import AfterValidator, ConfigDict, Field, field_validator, model_validator
 from typing_extensions import TypeAlias, TypeGuard
 
@@ -19,14 +18,7 @@
     validate_cron_string,
     validate_rrule_string,
 )
-
-if TYPE_CHECKING:
-    # type checkers have difficulty accepting that
-    # pydantic_extra_types.pendulum_dt and pendulum.DateTime can be used
-    # together.
-    DateTime = pendulum.DateTime
-else:
-    from prefect.types import DateTime
+from prefect.types._datetime import Date, DateTime
 
 MAX_ITERATIONS = 1000
 # approx. 1 years worth of RDATEs + buffer
@@ -83,7 +75,7 @@ class IntervalSchedule(PrefectBaseModel):
 
     interval: datetime.timedelta = Field(gt=datetime.timedelta(0))
     anchor_date: Annotated[DateTime, AfterValidator(default_anchor_date)] = Field(
-        default_factory=lambda: pendulum.now("UTC"),
+        default_factory=lambda: DateTime.now("UTC"),
         examples=["2020-01-01T00:00:00Z"],
     )
     timezone: Optional[str] = Field(default=None, examples=["America/New_York"])
@@ -99,9 +91,7 @@ def __init__(
             self,
             /,
             interval: datetime.timedelta,
-            anchor_date: Optional[
-                Union[pendulum.DateTime, datetime.datetime, str]
-            ] = None,
+            anchor_date: Optional[Union[DateTime, datetime.datetime, str]] = None,
             timezone: Optional[str] = None,
         ) -> None: ...
 
@@ -153,7 +143,7 @@ def valid_cron_string(cls, v: str) -> str:
         return validate_cron_string(v)
 
 
-DEFAULT_ANCHOR_DATE = pendulum.date(2020, 1, 1)
+DEFAULT_ANCHOR_DATE = Date(2020, 1, 1)
 
 
 def _rrule_dt(
@@ -219,7 +209,7 @@ def from_rrule(
             return RRuleSchedule(rrule=str(rrule), timezone=timezone)
         rrules = _rrule(rrule)
         dtstarts = [dts for rr in rrules if (dts := _rrule_dt(rr)) is not None]
-        unique_dstarts = set(pendulum.instance(d).in_tz("UTC") for d in dtstarts)
+        unique_dstarts = set(DateTime.instance(d).in_tz("UTC") for d in dtstarts)
         unique_timezones = set(d.tzinfo for d in dtstarts if d.tzinfo is not None)
 
         if len(unique_timezones) > 1:
@@ -380,7 +370,7 @@ def construct_schedule(
         if isinstance(interval, (int, float)):
             interval = datetime.timedelta(seconds=interval)
         if not anchor_date:
-            anchor_date = pendulum.DateTime.now()
+            anchor_date = DateTime.now()
         schedule = IntervalSchedule(
             interval=interval, anchor_date=anchor_date, timezone=timezone
         )