prefect-client 3.0.8__tar.gz → 3.0.10__tar.gz

{prefect-client-3.0.8/src/prefect_client.egg-info → prefect-client-3.0.10}/PKG-INFO

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

{prefect-client-3.0.8 → prefect-client-3.0.10}/setup.cfg

@@ -57,7 +57,7 @@ ignore_missing_imports = True
 
 [versioneer]
 VCS = git
-style = pep440-pre
+style = pep440
 versionfile_source = src/prefect/_version.py
 versionfile_build = prefect/_version.py
 version_regex = ^(\d+\.\d+\.\d+(?:[a-zA-Z0-9]+(?:\.[a-zA-Z0-9]+)*)?)$

{prefect-client-3.0.8 → prefect-client-3.0.10}/src/prefect/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2024-10-10T10:17:25-0500",
+ "date": "2024-10-15T13:31:59-0500",
  "dirty": true,
  "error": null,
- "full-revisionid": "0894bad49e256657cab8208205623c4c7b75ad1f",
- "version": "3.0.8"
+ "full-revisionid": "3aa2d89362c2fe8ee429f0c2cf7e623e34588029",
+ "version": "3.0.10"
 }
 '''  # END VERSION_JSON
 

{prefect-client-3.0.8 → prefect-client-3.0.10}/src/prefect/context.py

@@ -453,23 +453,6 @@ class SettingsContext(ContextModel):
     def __hash__(self) -> int:
         return hash(self.settings)
 
-    def __enter__(self):
-        """
-        Upon entrance, we ensure the home directory for the profile exists.
-        """
-        return_value = super().__enter__()
-
-        try:
-            prefect_home = self.settings.home
-            prefect_home.mkdir(mode=0o0700, exist_ok=True)
-        except OSError:
-            warnings.warn(
-                (f"Failed to create the Prefect home directory at {prefect_home}"),
-                stacklevel=2,
-            )
-
-        return return_value
-
     @classmethod
     def get(cls) -> "SettingsContext":
         # Return the global context instead of `None` if no context exists
@@ -567,9 +550,9 @@ def tags(*new_tags: str) -> Generator[Set[str], None, None]:
         {"a", "b", "c", "d", "e", "f"}
     """
     current_tags = TagsContext.get().current_tags
-    new_tags = current_tags.union(new_tags)
-    with TagsContext(current_tags=new_tags):
-        yield new_tags
+    _new_tags = current_tags.union(new_tags)
+    with TagsContext(current_tags=_new_tags):
+        yield _new_tags
 
 
 @contextmanager
@@ -659,7 +642,16 @@ def root_settings_context():
         )
         active_name = "ephemeral"
 
-    return SettingsContext(profile=profiles[active_name], settings=Settings())
+    if not (settings := Settings()).home.exists():
+        try:
+            settings.home.mkdir(mode=0o0700, exist_ok=True)
+        except OSError:
+            warnings.warn(
+                (f"Failed to create the Prefect home directory at {settings.home}"),
+                stacklevel=2,
+            )
+
+    return SettingsContext(profile=profiles[active_name], settings=settings)
 
     # Note the above context is exited and the global settings context is used by
     # an override in the `SettingsContext.get` method.

{prefect-client-3.0.8 → prefect-client-3.0.10}/src/prefect/flows.py

@@ -1245,7 +1245,7 @@ class Flow(Generic[P, R]):
     @overload
     def __call__(
         self: "Flow[P, Coroutine[Any, Any, T]]", *args: P.args, **kwargs: P.kwargs
-    ) -> Awaitable[T]:
+    ) -> Coroutine[Any, Any, T]:
         ...
 
     @overload

{prefect-client-3.0.8 → prefect-client-3.0.10}/src/prefect/runner/runner.py

@@ -398,10 +398,12 @@ class Runner:
         start_client_metrics_server()
 
         async with self as runner:
-            async with self._loops_task_group as tg:
+            # This task group isn't included in the exit stack because we want to
+            # stay in this function until the runner is told to stop
+            async with self._loops_task_group as loops_task_group:
                 for storage in self._storage_objs:
                     if storage.pull_interval:
-                        tg.start_soon(
+                        loops_task_group.start_soon(
                             partial(
                                 critical_service_loop,
                                 workload=storage.pull_code,
@@ -411,8 +413,8 @@ class Runner:
                             )
                         )
                     else:
-                        tg.start_soon(storage.pull_code)
-                tg.start_soon(
+                        loops_task_group.start_soon(storage.pull_code)
+                loops_task_group.start_soon(
                     partial(
                         critical_service_loop,
                         workload=runner._get_and_submit_flow_runs,
@@ -421,7 +423,7 @@ class Runner:
                         jitter_range=0.3,
                     )
                 )
-                tg.start_soon(
+                loops_task_group.start_soon(
                     partial(
                         critical_service_loop,
                         workload=runner._check_for_cancelled_flow_runs,
@@ -1264,15 +1266,15 @@ class Runner:
         if not hasattr(self, "_loop") or not self._loop:
             self._loop = asyncio.get_event_loop()
 
+        await self._client.__aenter__()
+
         if not hasattr(self, "_runs_task_group") or not self._runs_task_group:
             self._runs_task_group: anyio.abc.TaskGroup = anyio.create_task_group()
+        await self._runs_task_group.__aenter__()
 
         if not hasattr(self, "_loops_task_group") or not self._loops_task_group:
             self._loops_task_group: anyio.abc.TaskGroup = anyio.create_task_group()
 
-        await self._client.__aenter__()
-        await self._runs_task_group.__aenter__()
-
         self.started = True
         return self
 
@@ -1281,13 +1283,18 @@ class Runner:
         if self.pause_on_shutdown:
             await self._pause_schedules()
         self.started = False
+
         for scope in self._scheduled_task_scopes:
             scope.cancel()
+
         if self._runs_task_group:
             await self._runs_task_group.__aexit__(*exc_info)
+
         if self._client:
             await self._client.__aexit__(*exc_info)
+
         shutil.rmtree(str(self._tmp_dir))
+        del self._runs_task_group, self._loops_task_group
 
     def __repr__(self):
         return f"Runner(name={self.name!r})"