prefect-client 3.4.18.dev3__tar.gz → 3.4.18.dev5__tar.gz

{prefect_client-3.4.18.dev3 → prefect_client-3.4.18.dev5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prefect-client
-Version: 3.4.18.dev3
+Version: 3.4.18.dev5
 Summary: Workflow orchestration and management.
 Project-URL: Changelog, https://github.com/PrefectHQ/prefect/releases
 Project-URL: Documentation, https://docs.prefect.io

prefect_client-3.4.18.dev5/src/prefect/_build_info.py

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.4.18.dev5"
+__build_date__ = "2025-09-12 08:09:10.791118+00:00"
+__git_commit__ = "e4ef2c364a44968a8cded844e0d29e6c5fd8db81"
+__dirty__ = False

{prefect_client-3.4.18.dev3 → prefect_client-3.4.18.dev5}/src/prefect/runner/runner.py

@@ -1488,7 +1488,8 @@ class Runner:
     async def __aenter__(self) -> Self:
         self._logger.debug("Starting runner...")
         self._client = get_client()
-        self._tmp_dir.mkdir(parents=True)
+        # Be tolerant to concurrent/duplicate initialization attempts
+        self._tmp_dir.mkdir(parents=True, exist_ok=True)
 
         self._limiter = anyio.CapacityLimiter(self.limit) if self.limit else None
 
@@ -1535,7 +1536,8 @@ class Runner:
 
         await self._exit_stack.__aexit__(*exc_info)
 
-        shutil.rmtree(str(self._tmp_dir))
+        # Be tolerant to already-removed temp directories
+        shutil.rmtree(str(self._tmp_dir), ignore_errors=True)
         del self._runs_task_group, self._loops_task_group
 
         if self._heartbeat_task:

{prefect_client-3.4.18.dev3 → prefect_client-3.4.18.dev5}/src/prefect/workers/base.py

@@ -679,6 +679,16 @@ class BaseWorker(abc.ABC, Generic[C, V, R]):
                         )
                         healthcheck_thread.start()
                     printer(f"Worker {worker.name!r} started!")
+
+                # If running once, wait for active runs to finish before teardown
+                if run_once and self._limiter:
+                    # Use the limiter's borrowed token count as the source of truth
+                    while self.limiter.borrowed_tokens > 0:
+                        self._logger.debug(
+                            "Waiting for %s active run(s) to finish before shutdown...",
+                            self.limiter.borrowed_tokens,
+                        )
+                        await anyio.sleep(0.1)
         finally:
             stop_client_metrics_server()
 

{prefect_client-3.4.18.dev3 → prefect_client-3.4.18.dev5}/src/prefect/workers/process.py

@@ -202,6 +202,15 @@ class ProcessWorker(
                         )
                         healthcheck_thread.start()
                     printer(f"Worker {worker.name!r} started!")
+
+                # If running once, wait for active runs to complete before exiting
+                if run_once and self._limiter:
+                    while self.limiter.borrowed_tokens > 0:
+                        self._logger.debug(
+                            "Waiting for %s active run(s) to finish before shutdown...",
+                            self.limiter.borrowed_tokens,
+                        )
+                        await anyio.sleep(0.1)
         finally:
             stop_client_metrics_server()
 

prefect_client-3.4.18.dev3/src/prefect/_build_info.py

@@ -1,5 +0,0 @@
-# Generated by versioningit
-__version__ = "3.4.18.dev3"
-__build_date__ = "2025-09-10 08:09:19.131681+00:00"
-__git_commit__ = "40d354fe1837259ba451a2e8cd345aa5f58eb864"
-__dirty__ = False