prefect-client 3.0.0rc12__py3-none-any.whl → 3.0.0rc13__py3-none-any.whl

prefect/events/clients.py

@@ -140,6 +140,11 @@ class AssertingEventsClient(EventsClient):
         cls.last = None
         cls.all = []
 
+    def pop_events(self) -> List[Event]:
+        events = self.events
+        self.events = []
+        return events
+
     async def _emit(self, event: Event) -> None:
         self.events.append(event)
 

prefect/settings.py

@@ -1383,11 +1383,6 @@ PREFECT_WORKER_WEBSERVER_PORT = Setting(
 The port the worker's webserver should bind to.
 """
 
-PREFECT_API_SERVICES_TASK_SCHEDULING_ENABLED = Setting(bool, default=True)
-"""
-Whether or not to start the task scheduling service in the server application.
-"""
-
 PREFECT_TASK_SCHEDULING_DEFAULT_STORAGE_BLOCK = Setting(Optional[str], default=None)
 """The `block-type/block-document` slug of a block to use as the default storage
 for autonomous tasks."""

prefect/task_engine.py

@@ -577,19 +577,23 @@ class TaskRunEngine(Generic[P, R]):
                 self._client = client_ctx.sync_client
                 self._is_started = True
                 try:
-                    if not self.task_run:
-                        if PREFECT_EXPERIMENTAL_ENABLE_CLIENT_SIDE_TASK_ORCHESTRATION:
-                            # TODO - this maybe should be a method on Task?
-                            from prefect.utilities.engine import (
-                                _resolve_custom_task_run_name,
+                    if PREFECT_EXPERIMENTAL_ENABLE_CLIENT_SIDE_TASK_ORCHESTRATION:
+                        from prefect.utilities.engine import (
+                            _resolve_custom_task_run_name,
+                        )
+
+                        task_run_name = (
+                            _resolve_custom_task_run_name(
+                                task=self.task, parameters=self.parameters
                             )
+                            if self.task.task_run_name
+                            else None
+                        )
 
-                            task_run_name = None
-                            if not self._task_name_set and self.task.task_run_name:
-                                task_run_name = _resolve_custom_task_run_name(
-                                    task=self.task, parameters=self.parameters
-                                )
+                        if self.task_run and task_run_name:
+                            self.task_run.name = task_run_name
 
+                        if not self.task_run:
                             self.task_run = run_coro_as_sync(
                                 self.task.create_local_run(
                                     id=task_run_id,
@@ -601,7 +605,8 @@ class TaskRunEngine(Generic[P, R]):
                                     task_run_name=task_run_name,
                                 )
                             )
-                        else:
+                    else:
+                        if not self.task_run:
                             self.task_run = run_coro_as_sync(
                                 self.task.create_run(
                                     id=task_run_id,

prefect/task_worker.py

@@ -29,6 +29,7 @@ from prefect.logging.loggers import get_logger
 from prefect.results import ResultFactory
 from prefect.settings import (
     PREFECT_API_URL,
+    PREFECT_EXPERIMENTAL_ENABLE_CLIENT_SIDE_TASK_ORCHESTRATION,
     PREFECT_TASK_SCHEDULING_DELETE_FAILED_SUBMISSIONS,
 )
 from prefect.states import Pending
@@ -289,26 +290,33 @@ class TaskWorker:
                     await self._client._client.delete(f"/task_runs/{task_run.id}")
                 return
 
-        try:
+        if PREFECT_EXPERIMENTAL_ENABLE_CLIENT_SIDE_TASK_ORCHESTRATION:
             new_state = Pending()
             new_state.state_details.deferred = True
-            state = await propose_state(
-                client=get_client(),  # TODO prove that we cannot use self._client here
-                state=new_state,
-                task_run_id=task_run.id,
-            )
-        except Abort as exc:
-            logger.exception(
-                f"Failed to submit task run {task_run.id!r} to engine", exc_info=exc
-            )
-            return
-        except PrefectHTTPStatusError as exc:
-            if exc.response.status_code == 404:
-                logger.warning(
-                    f"Task run {task_run.id!r} not found. It may have been deleted."
+            new_state.state_details.task_run_id = task_run.id
+            new_state.state_details.flow_run_id = task_run.flow_run_id
+            state = new_state
+        else:
+            try:
+                new_state = Pending()
+                new_state.state_details.deferred = True
+                state = await propose_state(
+                    client=get_client(),  # TODO prove that we cannot use self._client here
+                    state=new_state,
+                    task_run_id=task_run.id,
+                )
+            except Abort as exc:
+                logger.exception(
+                    f"Failed to submit task run {task_run.id!r} to engine", exc_info=exc
                 )
                 return
-            raise
+            except PrefectHTTPStatusError as exc:
+                if exc.response.status_code == 404:
+                    logger.warning(
+                        f"Task run {task_run.id!r} not found. It may have been deleted."
+                    )
+                    return
+                raise
 
         if not state.is_pending():
             logger.warning(

{prefect_client-3.0.0rc12.dist-info → prefect_client-3.0.0rc13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prefect-client
-Version: 3.0.0rc12
+Version: 3.0.0rc13
 Summary: Workflow orchestration and management.
 Home-page: https://www.prefect.io
 Author: Prefect Technologies, Inc.

{prefect_client-3.0.0rc12.dist-info → prefect_client-3.0.0rc13.dist-info}/RECORD

@@ -20,12 +20,12 @@ prefect/profiles.toml,sha256=Fs8hD_BdWHZgAijgk8pK_Zx-Pm-YFixqDIfEP6fM-qU,38
 prefect/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prefect/results.py,sha256=3mVkVWZn_VSQ9Pik79StNy113rB_SEiP83SdoUsFvTM,24635
 prefect/serializers.py,sha256=Lo41EM0_qGzcfB_63390Izeo3DdK6cY6VZfxa9hpSGQ,8712
-prefect/settings.py,sha256=Xmgy4S_zQWKiT7VEXiRtSjhjihrqK5feNkH3QVdDTMQ,69090
+prefect/settings.py,sha256=gBn4DPcXLpIEIyH3TiyKBI1_PUZc4RuMKpYWF0Bo6mQ,68927
 prefect/states.py,sha256=lw22xucH46cN9stkxiV9ByIvq689mH5iL3gErri-Y18,22207
-prefect/task_engine.py,sha256=GXWIDMNUl0--RecH9yc41mSSpCNnIFmapi8fMIFDsdQ,39911
+prefect/task_engine.py,sha256=W7P_FkWUD_YZXpqDoa5abZJMH0UvvBt4Tdw4wfyypmY,39989
 prefect/task_runners.py,sha256=W1n0yMwbDIqnvffFVJADo9MGEbLaYkzWk52rqgnkMY4,15019
 prefect/task_runs.py,sha256=eDWYH5H1K4SyduhKmn3GzO6vM3fZSwOZxAb8KhkMGsk,7798
-prefect/task_worker.py,sha256=lV9rQb9YOaO28DZLW_avw6p0pTSVYtsA1gqODWxB7J0,17334
+prefect/task_worker.py,sha256=SJaetUGQQDOQItLPnQ7vDgAX7lAIsOcm5zSV0jmYA3M,17812
 prefect/tasks.py,sha256=A-sQB8S5dxcLtl10crDThsfdKpMzm9Rknfwhu27rnvY,68085
 prefect/transactions.py,sha256=XBbOjAUnDWw9QcxVwEamRaWxvRA_Ao-MkIN5dFL7h54,10008
 prefect/variables.py,sha256=-t5LVY0N-K4f0fa6YwruVVQqwnU3fGWBMYXXE32XPkA,4821
@@ -104,7 +104,7 @@ prefect/docker/__init__.py,sha256=jumlacz2HY9l1ee0L9_kE0PFi9NO3l3pWINm9T5N9hs,52
 prefect/docker/docker_image.py,sha256=Y84_ooCYA9NGl6FElJul9-FaW3teT-eia2SiNtZ1LG8,2999
 prefect/events/__init__.py,sha256=GtKl2bE--pJduTxelH2xy7SadlLJmmis8WR1EYixhuA,2094
 prefect/events/actions.py,sha256=4kBV2NwFlC6oXVeMp9Qb2HMNqv1IZ7FcOqeXz1zlRf0,8983
-prefect/events/clients.py,sha256=nDP8AQCoPNOfRPgS9QbEvdpx0wKyE4_4gwxQEtsnuUA,19860
+prefect/events/clients.py,sha256=52jXK9MD8OGy9vFB74I0efAk7IGtlbiE0n5iADD-b7g,19978
 prefect/events/filters.py,sha256=IJ1TF-TCC7Wk2nJsbYW-HyAANToDQ6z1MdD63qE-lfw,8186
 prefect/events/related.py,sha256=1rUnQ7tg_UtNfSAkKdRo-rD2W93EKKB9xafPxyusFj8,6841
 prefect/events/utilities.py,sha256=gia_jGwxykxRTzS6FAp-gVEP9d7gH8S_hTd3-RQNJVQ,2627
@@ -186,8 +186,8 @@ prefect/workers/cloud.py,sha256=BOVVY5z-vUIQ2u8LwMTXDaNys2fjOZSS5YGDwJmTQjI,230
 prefect/workers/process.py,sha256=t1f1EYRoPL5B25KbLgUX2b5q-lCCAXb2Gpf6T2M9WfY,19822
 prefect/workers/server.py,sha256=lgh2FfSuaNU7b6HPxSFm8JtKvAvHsZGkiOo4y4tW1Cw,2022
 prefect/workers/utilities.py,sha256=VfPfAlGtTuDj0-Kb8WlMgAuOfgXCdrGAnKMapPSBrwc,2483
-prefect_client-3.0.0rc12.dist-info/LICENSE,sha256=MCxsn8osAkzfxKC4CC_dLcUkU8DZLkyihZ8mGs3Ah3Q,11357
-prefect_client-3.0.0rc12.dist-info/METADATA,sha256=mpzyZ76ECL9XsRq4Squ61UkfenYkdhJzZM7FekCZjHg,7432
-prefect_client-3.0.0rc12.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-prefect_client-3.0.0rc12.dist-info/top_level.txt,sha256=MJZYJgFdbRc2woQCeB4vM6T33tr01TmkEhRcns6H_H4,8
-prefect_client-3.0.0rc12.dist-info/RECORD,,
+prefect_client-3.0.0rc13.dist-info/LICENSE,sha256=MCxsn8osAkzfxKC4CC_dLcUkU8DZLkyihZ8mGs3Ah3Q,11357
+prefect_client-3.0.0rc13.dist-info/METADATA,sha256=s2r3T5XgBq13oTAtYVrA5ADAs9zvB3Ls5rAwubtA9H8,7432
+prefect_client-3.0.0rc13.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+prefect_client-3.0.0rc13.dist-info/top_level.txt,sha256=MJZYJgFdbRc2woQCeB4vM6T33tr01TmkEhRcns6H_H4,8
+prefect_client-3.0.0rc13.dist-info/RECORD,,