prefect-client 2.20.4__tar.gz → 2.20.6__tar.gz

{prefect_client-2.20.4/src/prefect_client.egg-info → prefect-client-2.20.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prefect-client
-Version: 2.20.4
+Version: 2.20.6
 Summary: Workflow orchestration and management.
 Home-page: https://www.prefect.io
 Author: Prefect Technologies, Inc.
@@ -21,46 +21,8 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Topic :: Software Development :: Libraries
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: anyio<5.0.0,>=4.4.0
-Requires-Dist: asgi-lifespan<3.0,>=1.0
-Requires-Dist: cachetools<6.0,>=5.3
-Requires-Dist: cloudpickle<4.0,>=2.0
-Requires-Dist: coolname<3.0.0,>=1.0.4
-Requires-Dist: croniter<3.0.0,>=1.0.12
-Requires-Dist: fsspec>=2022.5.0
-Requires-Dist: exceptiongroup>=1.2.1
-Requires-Dist: graphviz>=0.20.1
-Requires-Dist: griffe<2.0.0,>=0.49.0
-Requires-Dist: httpcore<2.0.0,>=1.0.5
-Requires-Dist: httpx[http2]!=0.23.2,>=0.23
-Requires-Dist: importlib_metadata>=4.4; python_version < "3.10"
-Requires-Dist: importlib-resources<6.5.0,>=6.1.3
-Requires-Dist: jsonpatch<2.0,>=1.32
-Requires-Dist: jsonschema<5.0.0,>=4.0.0
-Requires-Dist: orjson<4.0,>=3.7
-Requires-Dist: packaging<24.3,>=21.3
-Requires-Dist: pathspec>=0.8.0
-Requires-Dist: pendulum<3.0; python_version < "3.12"
-Requires-Dist: pendulum<4,>=3.0.0; python_version >= "3.12"
-Requires-Dist: pydantic[email]!=2.0.0,!=2.0.1,!=2.1.0,<3.0.0,>=1.10.0
-Requires-Dist: pydantic_core<3.0.0,>=2.12.0
-Requires-Dist: python_dateutil<3.0.0,>=2.8.2
-Requires-Dist: python-slugify<9.0,>=5.0
-Requires-Dist: pyyaml<7.0.0,>=5.4.1
-Requires-Dist: rfc3339-validator<0.2.0,>=0.1.4
-Requires-Dist: rich<14.0,>=11.0
-Requires-Dist: ruamel.yaml>=0.17.0
-Requires-Dist: sniffio<2.0.0,>=1.3.0
-Requires-Dist: toml>=0.10.0
-Requires-Dist: typing_extensions<5.0.0,>=4.5.0
-Requires-Dist: ujson<6.0.0,>=5.8.0
-Requires-Dist: uvicorn!=0.29.0,>=0.14.0
-Requires-Dist: websockets<14.0,>=10.4
-Requires-Dist: itsdangerous
-Requires-Dist: python-multipart>=0.0.7
 Provides-Extra: notifications
-Requires-Dist: apprise<2.0.0,>=1.1.0; extra == "notifications"
+License-File: LICENSE
 
 <p align="center"><img src="https://github.com/PrefectHQ/prefect/assets/3407835/c654cbc6-63e8-4ada-a92a-efd2f8f24b85" width=1000></p>
 

prefect-client-2.20.6/src/prefect/_internal/integrations.py

@@ -0,0 +1,7 @@
+KNOWN_EXTRAS_FOR_PACKAGES = {
+    "prefect-kubernetes": "prefect[kubernetes]",
+    "prefect-aws": "prefect[aws]",
+    "prefect-gcp": "prefect[gcp]",
+    "prefect-azure": "prefect[azure]",
+    "prefect-docker": "prefect[docker]",
+}

{prefect_client-2.20.4 → prefect-client-2.20.6}/src/prefect/blocks/core.py

@@ -507,10 +507,9 @@ class Block(BaseModel, ABC):
         `<module>:11: No type or annotation for parameter 'write_json'`
         because griffe is unable to parse the types from pydantic.BaseModel.
         """
-        with disable_logger("griffe.docstrings.google"):
-            with disable_logger("griffe.agents.nodes"):
-                docstring = Docstring(cls.__doc__)
-                parsed = parse(docstring, Parser.google)
+        with disable_logger("griffe"):
+            docstring = Docstring(cls.__doc__)
+            parsed = parse(docstring, Parser.google)
         return parsed
 
     @classmethod

{prefect_client-2.20.4 → prefect-client-2.20.6}/src/prefect/deployments/steps/core.py

@@ -21,6 +21,7 @@ from typing import Any, Dict, List, Optional, Tuple, Union
 
 from prefect._internal.compatibility.deprecated import PrefectDeprecationWarning
 from prefect._internal.concurrency.api import Call, from_async
+from prefect._internal.integrations import KNOWN_EXTRAS_FOR_PACKAGES
 from prefect.logging.loggers import get_logger
 from prefect.settings import PREFECT_DEBUG_MODE
 from prefect.utilities.importtools import import_object
@@ -83,6 +84,11 @@ def _get_function_for_step(
             raise
 
     try:
+        packages = [
+            KNOWN_EXTRAS_FOR_PACKAGES.get(package, package)
+            for package in packages
+            if package
+        ]
         subprocess.check_call([sys.executable, "-m", "pip", "install", *packages])
     except subprocess.CalledProcessError:
         get_logger("deployments.steps.core").warning(

{prefect_client-2.20.4 → prefect-client-2.20.6}/src/prefect/tasks.py

@@ -141,8 +141,17 @@ def _generate_task_key(fn: Callable[..., Any]) -> str:
 
     qualname = fn.__qualname__.split(".")[-1]
 
+    try:
+        code_obj = getattr(fn, "__code__", None)
+        if code_obj is None:
+            code_obj = fn.__call__.__code__
+    except AttributeError:
+        raise AttributeError(
+            f"{fn} is not a standard Python function object and could not be converted to a task."
+        ) from None
+
     code_hash = (
-        h[:NUM_CHARS_DYNAMIC_KEY] if (h := hash_objects(fn.__code__)) else "unknown"
+        h[:NUM_CHARS_DYNAMIC_KEY] if (h := hash_objects(code_obj)) else "unknown"
     )
 
     return f"{qualname}-{code_hash}"

{prefect_client-2.20.4 → prefect-client-2.20.6}/src/prefect/utilities/callables.py

@@ -255,9 +255,7 @@ def parameter_docstrings(docstring: Optional[str]) -> Dict[str, str]:
     if not docstring:
         return param_docstrings
 
-    with disable_logger("griffe.docstrings.google"), disable_logger(
-        "griffe.agents.nodes"
-    ):
+    with disable_logger("griffe"):
         parsed = parse(Docstring(docstring), Parser.google)
         for section in parsed:
             if section.kind != DocstringSectionKind.parameters:

{prefect_client-2.20.4 → prefect-client-2.20.6}/src/prefect/variables.py

@@ -1,4 +1,4 @@
-from typing import List, Optional
+from typing import List, Optional, Union
 
 from prefect._internal.compatibility.deprecated import deprecated_callable
 from prefect.client.schemas.actions import VariableCreate as VariableRequest
@@ -26,7 +26,7 @@ class Variable(VariableRequest):
         value: str,
         tags: Optional[List[str]] = None,
         overwrite: bool = False,
-    ) -> Optional[str]:
+    ) -> Optional["Variable"]:
         """
         Sets a new variable. If one exists with the same name, user must pass `overwrite=True`
         ```
@@ -47,8 +47,7 @@ class Variable(VariableRequest):
         """
         client, _ = get_or_create_client()
         variable = await client.read_variable_by_name(name)
-        var_dict = {"name": name, "value": value}
-        var_dict["tags"] = tags or []
+        var_dict = {"name": name, "value": value, "tags": tags or []}
         if variable:
             if not overwrite:
                 raise ValueError(
@@ -65,7 +64,9 @@ class Variable(VariableRequest):
 
     @classmethod
     @sync_compatible
-    async def get(cls, name: str, default: Optional[str] = None) -> Optional[str]:
+    async def get(
+        cls, name: str, default: Optional[str] = None
+    ) -> Optional[Union["Variable", str]]:
         """
         Get a variable by name. If doesn't exist return the default.
         ```
@@ -86,7 +87,12 @@ class Variable(VariableRequest):
         """
         client, _ = get_or_create_client()
         variable = await client.read_variable_by_name(name)
-        return variable if variable else default
+
+        return (
+            Variable(name=variable.name, value=variable.value, tags=variable.tags)
+            if variable
+            else default
+        )
 
 
 @deprecated_callable(start_date="Apr 2024")
@@ -110,5 +116,7 @@ async def get(name: str, default: Optional[str] = None) -> Optional[str]:
             var = await variables.get("my_var")
     ```
     """
-    variable = await Variable.get(name)
-    return variable.value if variable else default
+    variable = await Variable.get(name, default=default)
+    if isinstance(variable, Variable):
+        variable = variable.value
+    return variable

{prefect_client-2.20.4 → prefect-client-2.20.6/src/prefect_client.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prefect-client
-Version: 2.20.4
+Version: 2.20.6
 Summary: Workflow orchestration and management.
 Home-page: https://www.prefect.io
 Author: Prefect Technologies, Inc.
@@ -21,46 +21,8 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Topic :: Software Development :: Libraries
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: anyio<5.0.0,>=4.4.0
-Requires-Dist: asgi-lifespan<3.0,>=1.0
-Requires-Dist: cachetools<6.0,>=5.3
-Requires-Dist: cloudpickle<4.0,>=2.0
-Requires-Dist: coolname<3.0.0,>=1.0.4
-Requires-Dist: croniter<3.0.0,>=1.0.12
-Requires-Dist: fsspec>=2022.5.0
-Requires-Dist: exceptiongroup>=1.2.1
-Requires-Dist: graphviz>=0.20.1
-Requires-Dist: griffe<2.0.0,>=0.49.0
-Requires-Dist: httpcore<2.0.0,>=1.0.5
-Requires-Dist: httpx[http2]!=0.23.2,>=0.23
-Requires-Dist: importlib_metadata>=4.4; python_version < "3.10"
-Requires-Dist: importlib-resources<6.5.0,>=6.1.3
-Requires-Dist: jsonpatch<2.0,>=1.32
-Requires-Dist: jsonschema<5.0.0,>=4.0.0
-Requires-Dist: orjson<4.0,>=3.7
-Requires-Dist: packaging<24.3,>=21.3
-Requires-Dist: pathspec>=0.8.0
-Requires-Dist: pendulum<3.0; python_version < "3.12"
-Requires-Dist: pendulum<4,>=3.0.0; python_version >= "3.12"
-Requires-Dist: pydantic[email]!=2.0.0,!=2.0.1,!=2.1.0,<3.0.0,>=1.10.0
-Requires-Dist: pydantic_core<3.0.0,>=2.12.0
-Requires-Dist: python_dateutil<3.0.0,>=2.8.2
-Requires-Dist: python-slugify<9.0,>=5.0
-Requires-Dist: pyyaml<7.0.0,>=5.4.1
-Requires-Dist: rfc3339-validator<0.2.0,>=0.1.4
-Requires-Dist: rich<14.0,>=11.0
-Requires-Dist: ruamel.yaml>=0.17.0
-Requires-Dist: sniffio<2.0.0,>=1.3.0
-Requires-Dist: toml>=0.10.0
-Requires-Dist: typing_extensions<5.0.0,>=4.5.0
-Requires-Dist: ujson<6.0.0,>=5.8.0
-Requires-Dist: uvicorn!=0.29.0,>=0.14.0
-Requires-Dist: websockets<14.0,>=10.4
-Requires-Dist: itsdangerous
-Requires-Dist: python-multipart>=0.0.7
 Provides-Extra: notifications
-Requires-Dist: apprise<2.0.0,>=1.1.0; extra == "notifications"
+License-File: LICENSE
 
 <p align="center"><img src="https://github.com/PrefectHQ/prefect/assets/3407835/c654cbc6-63e8-4ada-a92a-efd2f8f24b85" width=1000></p>
 

{prefect_client-2.20.4 → prefect-client-2.20.6}/src/prefect_client.egg-info/SOURCES.txt

@@ -37,6 +37,7 @@ src/prefect/tasks.py
 src/prefect/variables.py
 src/prefect/_internal/__init__.py
 src/prefect/_internal/_logging.py
+src/prefect/_internal/integrations.py
 src/prefect/_internal/pytz.py
 src/prefect/_internal/compatibility/__init__.py
 src/prefect/_internal/compatibility/deprecated.py
@@ -94,7 +95,6 @@ src/prefect/_vendor/fastapi/exceptions.py
 src/prefect/_vendor/fastapi/logger.py
 src/prefect/_vendor/fastapi/param_functions.py
 src/prefect/_vendor/fastapi/params.py
-src/prefect/_vendor/fastapi/py.typed
 src/prefect/_vendor/fastapi/requests.py
 src/prefect/_vendor/fastapi/responses.py
 src/prefect/_vendor/fastapi/routing.py
@@ -140,7 +140,6 @@ src/prefect/_vendor/starlette/datastructures.py
 src/prefect/_vendor/starlette/endpoints.py
 src/prefect/_vendor/starlette/exceptions.py
 src/prefect/_vendor/starlette/formparsers.py
-src/prefect/_vendor/starlette/py.typed
 src/prefect/_vendor/starlette/requests.py
 src/prefect/_vendor/starlette/responses.py
 src/prefect/_vendor/starlette/routing.py
@@ -300,30 +299,4 @@ src/prefect_client.egg-info/PKG-INFO
 src/prefect_client.egg-info/SOURCES.txt
 src/prefect_client.egg-info/dependency_links.txt
 src/prefect_client.egg-info/requires.txt
-src/prefect_client.egg-info/top_level.txt
-tests/test_artifacts.py
-tests/test_automations.py
-tests/test_background_tasks.py
-tests/test_context.py
-tests/test_deployments.py
-tests/test_engine.py
-tests/test_exceptions.py
-tests/test_filesystems.py
-tests/test_flow_runs.py
-tests/test_flows.py
-tests/test_flows_compat.py
-tests/test_futures.py
-tests/test_highlighters.py
-tests/test_import_interceptor.py
-tests/test_log_prints.py
-tests/test_logging.py
-tests/test_new_flow_engine.py
-tests/test_new_task_engine.py
-tests/test_plugins.py
-tests/test_serializers.py
-tests/test_settings.py
-tests/test_states.py
-tests/test_task_runners.py
-tests/test_task_server.py
-tests/test_tasks.py
-tests/test_variables.py
+src/prefect_client.egg-info/top_level.txt