prefect-client 2.19.4__tar.gz → 3.0.0rc2__tar.gz

prefect-client-3.0.0rc2/PKG-INFO

@@ -0,0 +1,139 @@
+Metadata-Version: 2.1
+Name: prefect-client
+Version: 3.0.0rc2
+Summary: Workflow orchestration and management.
+Home-page: https://www.prefect.io
+Author: Prefect Technologies, Inc.
+Author-email: help@prefect.io
+License: UNKNOWN
+Project-URL: Changelog, https://github.com/PrefectHQ/prefect/blob/main/RELEASE-NOTES.md
+Project-URL: Documentation, https://docs.prefect.io
+Project-URL: Source, https://github.com/PrefectHQ/prefect
+Project-URL: Tracker, https://github.com/PrefectHQ/prefect/issues
+Platform: UNKNOWN
+Classifier: Natural Language :: English
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: System Administrators
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Provides-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>
+
+<p align="center">
+    <a href="https://pypi.python.org/pypi/prefect-client/" alt="PyPI version">
+        <img alt="PyPI" src="https://img.shields.io/pypi/v/prefect-client?color=0052FF&labelColor=090422"></a>
+    <a href="https://github.com/prefecthq/prefect/" alt="Stars">
+        <img src="https://img.shields.io/github/stars/prefecthq/prefect?color=0052FF&labelColor=090422" /></a>
+    <a href="https://pepy.tech/badge/prefect-client/" alt="Downloads">
+        <img src="https://img.shields.io/pypi/dm/prefect-client?color=0052FF&labelColor=090422" /></a>
+    <a href="https://github.com/prefecthq/prefect/pulse" alt="Activity">
+        <img src="https://img.shields.io/github/commit-activity/m/prefecthq/prefect?color=0052FF&labelColor=090422" /></a>
+    <br>
+    <a href="https://prefect.io/slack" alt="Slack">
+        <img src="https://img.shields.io/badge/slack-join_community-red.svg?color=0052FF&labelColor=090422&logo=slack" /></a>
+    <a href="https://discourse.prefect.io/" alt="Discourse">
+        <img src="https://img.shields.io/badge/discourse-browse_forum-red.svg?color=0052FF&labelColor=090422&logo=discourse" /></a>
+    <a href="https://www.youtube.com/c/PrefectIO/" alt="YouTube">
+        <img src="https://img.shields.io/badge/youtube-watch_videos-red.svg?color=0052FF&labelColor=090422&logo=youtube" /></a>
+</p>
+
+# prefect-client
+
+The `prefect-client` package is a minimal-installation of `prefect` which is designed for interacting with Prefect Cloud
+or remote any `prefect` server. It sheds some functionality and dependencies in exchange for a smaller installation size,
+making it ideal for use in lightweight or ephemeral environments. These characteristics make it ideal for use in lambdas
+or other resource-constrained environments.
+
+
+## Getting started
+
+`prefect-client` shares the same installation requirements as prefect. To install, make sure you are on Python 3.9 or
+later and run the following command:
+
+```bash
+pip install prefect-client
+```
+
+Next, ensure that your `prefect-client` has access to a remote `prefect` server by exporting the `PREFECT_API_KEY`
+(if using Prefect Cloud) and `PREFECT_API_URL` environment variables. Once those are set, use the package in your code as
+you would normally use `prefect`!
+
+
+For example, to remotely trigger a run a deployment:
+
+```python
+from prefect.deployments import run_deployment
+
+
+def my_lambda(event):
+    ...
+    run_deployment(
+        name="my-flow/my-deployment",
+        parameters={"foo": "bar"},
+        timeout=0,
+    )
+
+my_lambda({})
+```
+
+To emit events in an event driven system:
+
+```python
+from prefect.events import emit_event
+
+
+def something_happened():
+    emit_event("my-event", resource={"prefect.resource.id": "foo.bar"})
+
+something_happened()
+```
+
+
+Or just interact with a `prefect` API:
+```python
+from prefect.client.orchestration import get_client
+
+
+async def query_api():
+    async with get_client() as client:
+        limits = await client.read_concurrency_limits(limit=10, offset=0)
+        print(limits)
+
+
+query_api()
+```
+
+
+## Known limitations
+By design, `prefect-client` omits all CLI and server components. This means that the CLI is not available for use
+and attempts to access server objects will fail. Furthermore, some classes, methods, and objects may be available
+for import in `prefect-client` but may not be "runnable" if they tap into server-oriented functionality. If you
+encounter such a limitation, feel free to [open an issue](https://github.com/PrefectHQ/prefect/issues/new/choose)
+describing the functionality you are interested in using and we will do our best to make it available.
+
+
+## Next steps
+
+There's lots more you can do to orchestrate and observe your workflows with Prefect!
+Start with our [friendly tutorial](https://docs.prefect.io/tutorials) or explore the [core concepts of Prefect workflows](https://docs.prefect.io/concepts/).
+
+## Join the community
+
+Prefect is made possible by the fastest growing community of thousands of friendly data engineers. Join us in building a new kind of workflow system. The [Prefect Slack community](https://prefect.io/slack) is a fantastic place to learn more about Prefect, ask questions, or get help with workflow design. All community forums, including code contributions, issue discussions, and slack messages are subject to our [Code of Conduct](https://discourse.prefect.io/faq).
+
+## Contribute
+
+See our [documentation on contributing to Prefect](https://docs.prefect.io/contributing/overview/).
+
+Thanks for being part of the mission to build a new kind of workflow system and, of course, **happy engineering!**
+
+

{prefect-client-2.19.4 → prefect-client-3.0.0rc2}/README.md

@@ -20,24 +20,24 @@
 
 # prefect-client
 
-The `prefect-client` package is a minimal-installation of `prefect` which is designed for interacting with Prefect Cloud 
-or remote any `prefect` server. It sheds some functionality and dependencies in exchange for a smaller installation size, 
-making it ideal for use in lightweight or ephemeral environments. These characteristics make it ideal for use in lambdas 
-or other resource-constrained environments. 
+The `prefect-client` package is a minimal-installation of `prefect` which is designed for interacting with Prefect Cloud
+or remote any `prefect` server. It sheds some functionality and dependencies in exchange for a smaller installation size,
+making it ideal for use in lightweight or ephemeral environments. These characteristics make it ideal for use in lambdas
+or other resource-constrained environments.
 
 
 ## Getting started
 
-`prefect-client` shares the same installation requirements as prefect. To install, make sure you are on Python 3.8 or 
+`prefect-client` shares the same installation requirements as prefect. To install, make sure you are on Python 3.9 or
 later and run the following command:
 
 ```bash
 pip install prefect-client
 ```
 
-Next, ensure that your `prefect-client` has access to a remote `prefect` server by exporting the `PREFECT_API_KEY` 
-(if using Prefect Cloud) and `PREFECT_API_URL` environment variables. Once those are set, use the package in your code as 
-you would normally use `prefect`! 
+Next, ensure that your `prefect-client` has access to a remote `prefect` server by exporting the `PREFECT_API_KEY`
+(if using Prefect Cloud) and `PREFECT_API_URL` environment variables. Once those are set, use the package in your code as
+you would normally use `prefect`!
 
 
 For example, to remotely trigger a run a deployment:
@@ -86,10 +86,10 @@ query_api()
 
 
 ## Known limitations
-By design, `prefect-client` omits all CLI and server components. This means that the CLI is not available for use 
-and attempts to access server objects will fail. Furthermore, some classes, methods, and objects may be available 
-for import in `prefect-client` but may not be "runnable" if they tap into server-oriented functionality. If you 
-encounter such a limitation, feel free to [open an issue](https://github.com/PrefectHQ/prefect/issues/new/choose) 
+By design, `prefect-client` omits all CLI and server components. This means that the CLI is not available for use
+and attempts to access server objects will fail. Furthermore, some classes, methods, and objects may be available
+for import in `prefect-client` but may not be "runnable" if they tap into server-oriented functionality. If you
+encounter such a limitation, feel free to [open an issue](https://github.com/PrefectHQ/prefect/issues/new/choose)
 describing the functionality you are interested in using and we will do our best to make it available.
 
 

{prefect-client-2.19.4 → prefect-client-3.0.0rc2}/requirements-client.txt

@@ -1,9 +1,11 @@
-anyio >= 3.7.1, < 4.0.0
+anyio >= 4.0.0, < 5.0.0
 asgi-lifespan >= 1.0, < 3.0
 cachetools >= 5.3, < 6.0
 cloudpickle >= 2.0, < 4.0
 coolname >= 1.0.4, < 3.0.0
 croniter >= 1.0.12, < 3.0.0
+exceptiongroup >= 1.0.0
+fastapi >= 0.111.0, < 1.0.0
 fsspec >= 2022.5.0
 graphviz >= 0.20.1
 griffe >= 0.20.0
@@ -16,12 +18,12 @@ jsonschema >= 4.0.0, < 5.0.0
 orjson >= 3.7, < 4.0
 packaging >= 21.3, < 24.3
 pathspec >= 0.8.0
-# https://github.com/PrefectHQ/prefect/issues/11619
-pendulum < 3.0; python_version < '3.12'
-pendulum >= 3.0.0, <4; python_version >= '3.12'
-# the version constraints for pydantic are merged with those from fastapi 0.103.2
-pydantic[email]>=1.10.0,!=2.0.0,!=2.0.1,!=2.1.0,<3.0.0
+pendulum >= 3.0.0, <4
+# 2024/05/28: we had same as fastapi 0.103.2 -> pydantic 2.4 but for `Secret` we need 2.7
+pydantic >= 2.7, < 3.0.0
 pydantic_core >= 2.12.0, < 3.0.0
+pydantic_extra_types
+pydantic_settings
 python_dateutil >= 2.8.2, < 3.0.0
 python-slugify >= 5.0, < 9.0
 pyyaml >= 5.4.1, < 7.0.0
@@ -33,8 +35,4 @@ toml >= 0.10.0
 typing_extensions >= 4.5.0, < 5.0.0
 ujson >= 5.8.0, < 6.0.0
 uvicorn >= 0.14.0, < 0.29.0
-websockets >= 10.4, < 13.0
-
-# additional dependencies of starlette, which we're currently vendoring
-itsdangerous
-python-multipart>=0.0.7
+websockets >= 10.4, < 13.0

{prefect-client-2.19.4 → prefect-client-3.0.0rc2}/requirements-dev.txt

@@ -2,17 +2,13 @@ ruff
 cairosvg
 codespell>=2.2.6
 ddtrace
-# Pin IPython to 8.12 when building on Python 3.8
-# https://github.com/ipython/ipython/blob/main/README.rst
-ipython == 8.12.*; python_version == '3.8'
-ipython; python_version > '3.8'
+ipython
 jinja2
 mkdocs
 mkdocs-gen-files
 mkdocs-material
 mkdocstrings-python
 mike
-mock; python_version < '3.8'
 moto >= 5
 mypy >= 1.9.0
 numpy
@@ -29,8 +25,8 @@ pytest-timeout
 pytest-xdist < 3.4.0 # 3.5.0 introduces some flakiness. Need to investigate and resolve.
 pytkdocs >= 0.14.2
 pyyaml
-requests<2.32.0  # `requests` renamed `get_connection` to `_get_connection`, which is a method docker-py monkeypatches. See https://github.com/docker/docker-py/issues/3256 Should be able to un-pin when docker-py hotfix is out.
-setuptools != 60.9.0; python_version < '3.8'
+redis>=5.0.1
+setuptools
 vermin
 virtualenv
 watchfiles

{prefect-client-2.19.4 → prefect-client-3.0.0rc2}/requirements.txt

@@ -7,7 +7,7 @@ asyncpg >= 0.23
 click >= 8.0, < 8.2
 cryptography >= 36.0.1
 dateparser >= 1.1.1, < 2.0.0
-docker >= 4.0, < 7.0
+docker >= 4.0
 graphviz >= 0.20.1
 griffe >= 0.20.0
 jinja2 >= 3.0.0, < 4.0.0
@@ -16,5 +16,5 @@ humanize >= 4.9.0
 kubernetes >= 24.2.0, < 30.0.0
 pytz >= 2021.1, < 2025
 readchar >= 4.0.0, < 5.0.0
-sqlalchemy[asyncio] >= 1.4.22, != 1.4.33, < 3.0.0
+sqlalchemy[asyncio] >= 2.0, < 3.0.0
 typer >= 0.12.0, != 0.12.2, < 0.13.0

{prefect-client-2.19.4 → prefect-client-3.0.0rc2}/setup.cfg

@@ -1,4 +1,5 @@
 [tool:pytest]
+testpaths = tests
 addopts = -rfEs
 norecursedirs = *.egg-info .git .mypy_cache node_modules .pytest_cache .vscode
 python_files = 
@@ -18,15 +19,9 @@ asyncio_mode = auto
 timeout = 90
 filterwarnings = 
 	error
+	
 	ignore::DeprecationWarning:tornado.platform.asyncio.*
 	ignore::DeprecationWarning:tornado.ioloop
-	ignore:Support for class-based `config` is deprecated
-	ignore:The `__fields__` attribute is deprecated
-	ignore:The private method `_iter` will be removed
-	ignore:The `__fields_set__` attribute is deprecated, use `model_fields_set` instead
-	ignore:`__get_validators__` is deprecated and will be removed, use `__get_pydantic_core_schema__`
-	ignore:The `dict` method is deprecated; use `model_dump` instead
-	ignore:The `schema` method is deprecated; use `model_json_schema` instead
 	ignore:Default value default=
 	ignore:Implicitly cleaning up:ResourceWarning
 	ignore::ResourceWarning:distributed.diskutils
@@ -40,7 +35,6 @@ filterwarnings =
 	ignore:the imp module is deprecated:DeprecationWarning
 	ignore:distutils Version classes are deprecated:DeprecationWarning
 	ignore:The distutils package is deprecated:DeprecationWarning
-	ignore:Use setlocale:DeprecationWarning
 	ignore:Skipped unsupported reflection of expression-based index:sqlalchemy.exc.SAWarning
 	ignore:The --rsyncdir command line argument and rsyncdirs config variable are deprecated.:DeprecationWarning
 	ignore:Prefect will drop support for Python 3.7:FutureWarning
@@ -63,11 +57,11 @@ follow_imports = skip
 ignore_missing_imports = True
 
 [versioneer]
-vcs = git
+VCS = git
 style = pep440
 versionfile_source = src/prefect/_version.py
 versionfile_build = prefect/_version.py
-version_regex = ^(\d+\.\d+\.\d+)$
+version_regex = ^(\d+\.\d+\.\d+(?:[a-zA-Z0-9]+(?:\.[a-zA-Z0-9]+)*)?)$
 tag_prefix = 
 parentdir_prefix = 
 
@@ -88,8 +82,8 @@ backports =
 format = parsable
 eval_annotations = yes
 only_show_violations = yes
-targets = 3.8-
-exclusion_regex = ^src/prefect/_vendor/.*$|^src/prefect/utilities/compat\.py$
+targets = 3.9-
+exclusion_regex = ^src/prefect/_vendor/.*$|^src/prefect/utilities/compat\.py$|^tests/workers/test_process_worker.py$|^tests/test_background_tasks.py$|^src/integrations/.*/tests/.*$
 
 [egg_info]
 tag_build = 

{prefect-client-2.19.4 → prefect-client-3.0.0rc2}/setup.py

@@ -29,7 +29,7 @@ setup(
     package_dir={"": "src"},
     include_package_data=True,
     # Requirements
-    python_requires=">=3.8",
+    python_requires=">=3.9",
     install_requires=install_requires,
     extras_require={"notifications": ["apprise>=1.1.0, <2.0.0"]},
     classifiers=[
@@ -38,10 +38,10 @@ setup(
         "Intended Audience :: System Administrators",
         "License :: OSI Approved :: Apache Software License",
         "Programming Language :: Python :: 3 :: Only",
-        "Programming Language :: Python :: 3.8",
         "Programming Language :: Python :: 3.9",
         "Programming Language :: Python :: 3.10",
         "Programming Language :: Python :: 3.11",
+        "Programming Language :: Python :: 3.12",
         "Topic :: Software Development :: Libraries",
     ],
 )

{prefect-client-2.19.4 → prefect-client-3.0.0rc2}/src/prefect/__init__.py

@@ -5,8 +5,6 @@
 from . import _version
 import importlib
 import pathlib
-import warnings
-import sys
 
 __version_info__ = _version.get_versions()
 __version__ = __version_info__["version"]
@@ -31,12 +29,13 @@ from prefect.deployments import deploy
 from prefect.states import State
 from prefect.logging import get_run_logger
 from prefect.flows import flow, Flow, serve
+from prefect.transactions import Transaction
 from prefect.tasks import task, Task
 from prefect.context import tags
 from prefect.manifests import Manifest
 from prefect.utilities.annotations import unmapped, allow_failure
 from prefect.results import BaseResult
-from prefect.engine import pause_flow_run, resume_flow_run, suspend_flow_run
+from prefect.flow_runs import pause_flow_run, resume_flow_run, suspend_flow_run
 from prefect.client.orchestration import get_client, PrefectClient
 from prefect.client.cloud import get_cloud_client, CloudClient
 import prefect.variables
@@ -44,14 +43,9 @@ import prefect.runtime
 
 # Import modules that register types
 import prefect.serializers
-import prefect.deprecated.data_documents
-import prefect.deprecated.packaging
 import prefect.blocks.kubernetes
 import prefect.blocks.notifications
 import prefect.blocks.system
-import prefect.infrastructure.process
-import prefect.infrastructure.kubernetes
-import prefect.infrastructure.container
 
 # Initialize the process-wide profile and registry at import time
 import prefect.context
@@ -61,14 +55,11 @@ prefect.context.initialize_object_registry()
 # Perform any forward-ref updates needed for Pydantic models
 import prefect.client.schemas
 
-prefect.context.FlowRunContext.update_forward_refs(Flow=Flow)
-prefect.context.TaskRunContext.update_forward_refs(Task=Task)
-prefect.client.schemas.State.update_forward_refs(
-    BaseResult=BaseResult, DataDocument=prefect.deprecated.data_documents.DataDocument
-)
-prefect.client.schemas.StateCreate.update_forward_refs(
-    BaseResult=BaseResult, DataDocument=prefect.deprecated.data_documents.DataDocument
-)
+prefect.context.FlowRunContext.model_rebuild()
+prefect.context.TaskRunContext.model_rebuild()
+prefect.client.schemas.State.model_rebuild()
+prefect.client.schemas.StateCreate.model_rebuild()
+Transaction.model_rebuild()
 
 
 prefect.plugins.load_extra_entrypoints()
@@ -82,57 +73,17 @@ prefect.logging.get_logger("profiles").debug(
 )
 
 # Ensure moved names are accessible at old locations
-import prefect.client
-
 prefect.client.get_client = get_client
 prefect.client.PrefectClient = PrefectClient
 
 
 from prefect._internal.compatibility.deprecated import (
     inject_renamed_module_alias_finder,
-    register_renamed_module,
 )
 
-register_renamed_module(
-    "prefect.packaging", "prefect.deprecated.packaging", start_date="Mar 2024"
-)
 inject_renamed_module_alias_finder()
 
 
-# Attempt to warn users who are importing Prefect 1.x attributes that they may
-# have accidentally installed Prefect 2.x
-
-PREFECT_1_ATTRIBUTES = [
-    "prefect.Client",
-    "prefect.Parameter",
-    "prefect.api",
-    "prefect.apply_map",
-    "prefect.case",
-    "prefect.config",
-    "prefect.context",
-    "prefect.flatten",
-    "prefect.mapped",
-    "prefect.models",
-    "prefect.resource_manager",
-]
-
-
-class Prefect1ImportInterceptor(importlib.abc.Loader):
-    def find_spec(self, fullname, path, target=None):
-        if fullname in PREFECT_1_ATTRIBUTES:
-            warnings.warn(
-                f"Attempted import of {fullname!r}, which is part of Prefect 1.x, while"
-                f" Prefect {__version__} is installed. If you're upgrading you'll need"
-                " to update your code, see the Prefect 2.x migration guide:"
-                " `https://orion-docs.prefect.io/migration_guide/`. Otherwise ensure"
-                " that your code is pinned to the expected version."
-            )
-
-
-if not hasattr(sys, "frozen"):
-    sys.meta_path.insert(0, Prefect1ImportInterceptor())
-
-
 # Declare API for type-checkers
 __all__ = [
     "allow_failure",
@@ -145,6 +96,7 @@ __all__ = [
     "tags",
     "task",
     "Task",
+    "Transaction",
     "unmapped",
     "serve",
     "deploy",

{prefect-client-2.19.4 → prefect-client-3.0.0rc2}/src/prefect/_internal/compatibility/deprecated.py

@@ -9,21 +9,14 @@ Deprecated items require a start or end date. If a start date is given, the end
 will be calculated 6 months later. Start and end dates are always in the format MMM YYYY
 e.g. Jan 2023.
 """
+
 import functools
 import sys
 import warnings
-from typing import Any, Callable, Dict, List, Optional, Type, TypeVar, Union
+from typing import Any, Callable, List, Optional, Type, TypeVar
 
 import pendulum
-
-from prefect._internal.pydantic import HAS_PYDANTIC_V2
-
-if HAS_PYDANTIC_V2:
-    from pydantic.v1 import BaseModel, Field, root_validator
-    from pydantic.v1.schema import default_ref_template
-else:
-    from pydantic import BaseModel, Field, root_validator
-    from pydantic.schema import default_ref_template
+from pydantic import BaseModel
 
 from prefect.utilities.callables import get_call_parameters
 from prefect.utilities.importtools import (
@@ -237,9 +230,10 @@ def deprecated_field(
 
             cls_init(__pydantic_self__, **data)
 
-            field = __pydantic_self__.__fields__.get(name)
+            field = __pydantic_self__.model_fields.get(name)
             if field is not None:
-                field.field_info.extra["deprecated"] = True
+                field.json_schema_extra = field.json_schema_extra or {}
+                field.json_schema_extra["deprecated"] = True
 
         # Patch the model's init method
         model_cls.__init__ = __init__
@@ -278,106 +272,3 @@ def register_renamed_module(old_name: str, new_name: str, start_date: str):
     DEPRECATED_MODULE_ALIASES.append(
         AliasedModuleDefinition(old_name, new_name, callback)
     )
-
-
-class DeprecatedInfraOverridesField(BaseModel):
-    """
-    A model mixin that handles the deprecated `infra_overrides` field.
-
-    The `infra_overrides` field has been renamed to `job_variables`. This mixin maintains
-    backwards compatibility with users of the `infra_overrides` field while presenting
-    `job_variables` as the user-facing field.
-
-    When we remove support for `infra_overrides`, we can remove this class as a parent of
-    all schemas that use it, leaving them with only the `job_variables` field.
-    """
-
-    infra_overrides: Optional[Dict[str, Any]] = Field(
-        default_factory=dict,
-        description="Deprecated field. Use `job_variables` instead.",
-    )
-
-    @root_validator(pre=True)
-    def _job_variables_from_infra_overrides(
-        cls, values: Dict[str, Any]
-    ) -> Dict[str, Any]:
-        """
-        Validate that only one of `infra_overrides` or `job_variables` is used
-        and keep them in sync during init.
-        """
-        job_variables = values.get("job_variables")
-        infra_overrides = values.get("infra_overrides")
-
-        if job_variables is not None and infra_overrides is not None:
-            if job_variables != infra_overrides:
-                raise ValueError(
-                    "The `infra_overrides` field has been renamed to `job_variables`."
-                    "Use one of these fields, but not both."
-                )
-            return values
-        elif job_variables is not None and infra_overrides is None:
-            values["infra_overrides"] = job_variables
-        elif job_variables is None and infra_overrides is not None:
-            values["job_variables"] = infra_overrides
-        return values
-
-    def __setattr__(self, key: str, value: Any) -> None:
-        """
-        Override the default __setattr__ to ensure that setting `infra_overrides` or
-        `job_variables` will update both fields.
-        """
-        if key == "infra_overrides" or key == "job_variables":
-            updates = {"infra_overrides": value, "job_variables": value}
-            self.__dict__.update(updates)
-            return
-        super().__setattr__(key, value)
-
-    def dict(self, **kwargs) -> Dict[str, Any]:
-        """
-        Override the default dict method to ensure only `infra_overrides` is serialized.
-        This preserves backwards compatibility for newer clients talking to older servers.
-        """
-        exclude: Union[set, Dict[str, Any]] = kwargs.pop("exclude", set())
-        exclude_type = type(exclude)
-
-        if exclude_type is set:
-            exclude.add("job_variables")
-        elif exclude_type is dict:
-            exclude["job_variables"] = True
-        else:
-            exclude = {"job_variables"}
-        kwargs["exclude"] = exclude
-
-        return super().dict(**kwargs)
-
-    @classmethod
-    def schema(
-        cls, by_alias: bool = True, ref_template: str = default_ref_template
-    ) -> Dict[str, Any]:
-        """
-        Don't use the mixin docstring as the description if this class is missing a
-        docstring.
-        """
-        schema = super().schema(by_alias=by_alias, ref_template=ref_template)
-
-        if not cls.__doc__:
-            schema.pop("description", None)
-
-        return schema
-
-
-def handle_deprecated_infra_overrides_parameter(
-    job_variables: Dict[str, Any], infra_overrides: Dict[str, Any]
-) -> Optional[Dict[str, Any]]:
-    if infra_overrides is not None and job_variables is not None:
-        raise RuntimeError(
-            "The `infra_overrides` argument has been renamed to `job_variables`."
-            "Use one or the other, but not both."
-        )
-    elif infra_overrides is not None and job_variables is None:
-        jv = infra_overrides
-    elif job_variables is not None and infra_overrides is None:
-        jv = job_variables
-    else:
-        jv = None
-    return jv