Fix some tests for real this time

PrefectHQ · Feb 5, 2025 · 2664ecb · 2664ecb
1 parent ff16fa7
commit 2664ecb
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/prefect/deployments/runner.py b/src/prefect/deployments/runner.py
@@ -537,6 +537,7 @@ def from_flow(
             cron=cron,
             rrule=rrule,
             schedules=schedules,
+            schedule=schedule,
         )
 
         job_variables = job_variables or {}
@@ -631,6 +632,7 @@ def from_entrypoint(
         cron: Optional[Union[Iterable[str], str]] = None,
         rrule: Optional[Union[Iterable[str], str]] = None,
         paused: Optional[bool] = None,
+        schedule: Optional[Schedule] = None,
         schedules: Optional["FlexibleScheduleList"] = None,
         concurrency_limit: Optional[Union[int, ConcurrencyLimitConfig, None]] = None,
         parameters: Optional[dict[str, Any]] = None,
@@ -686,6 +688,7 @@ def from_entrypoint(
             cron=cron,
             rrule=rrule,
             schedules=schedules,
+            schedule=schedule,
         )
 
         if isinstance(concurrency_limit, ConcurrencyLimitConfig):
@@ -792,6 +795,7 @@ async def afrom_storage(
             cron=cron,
             rrule=rrule,
             schedules=schedules,
+            schedule=schedule,
         )
 
         if isinstance(concurrency_limit, ConcurrencyLimitConfig):
@@ -913,6 +917,7 @@ def from_storage(
             cron=cron,
             rrule=rrule,
             schedules=schedules,
+            schedule=schedule,
         )
 
         if isinstance(concurrency_limit, ConcurrencyLimitConfig):