prefect-client 2.20.17__tar.gz → 2.20.18__tar.gz

{prefect_client-2.20.17/src/prefect_client.egg-info → prefect_client-2.20.18}/PKG-INFO

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

{prefect_client-2.20.17 → prefect_client-2.20.18}/requirements-dev.txt

@@ -15,6 +15,8 @@ mock; python_version < '3.8'
 moto >= 5
 mypy >= 1.9.0
 numpy
+pendulum[test] < 3.0; python_version < '3.12'
+pendulum[test] >= 3.0.0, <4; python_version >= '3.12'
 pillow
 pre-commit
 pluggy >= 1.4.0

{prefect_client-2.20.17 → prefect_client-2.20.18}/src/prefect/utilities/callables.py

@@ -379,16 +379,17 @@ def generate_parameter_schema(
     model_fields = {}
     aliases = {}
 
-    class ModelConfig:
-        arbitrary_types_allowed = True
-
     if HAS_PYDANTIC_V2 and not has_v1_type_as_param(signature):
         create_schema = create_v2_schema
         process_params = process_v2_params
+        ModelConfig = pydantic.ConfigDict(arbitrary_types_allowed=True)
     else:
         create_schema = create_v1_schema
         process_params = process_v1_params
 
+        class ModelConfig:
+            arbitrary_types_allowed = True
+
     for position, param in enumerate(signature.parameters.values()):
         name, type_, field = process_params(
             param, position=position, docstrings=docstrings, aliases=aliases

{prefect_client-2.20.17 → prefect_client-2.20.18/src/prefect_client.egg-info}/PKG-INFO

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

{prefect_client-2.20.17 → prefect_client-2.20.18}/tests/test_plugins.py

@@ -83,20 +83,24 @@ def test_load_extra_entrypoints_strips_spaces():
 
 
 @pytest.mark.usefixtures("module_fixture")
-def test_load_extra_entrypoints_unparsable_entrypoint(capsys):
+def test_load_extra_entrypoints_unparsable_entrypoint(
+    capsys: pytest.CaptureFixture[str],
+):
     with temporary_settings({PREFECT_EXTRA_ENTRYPOINTS: "foo$bar"}):
         result = load_extra_entrypoints()
 
     assert set(result.keys()) == {"foo$bar"}
-    assert exceptions_equal(
-        result["foo$bar"], AttributeError("'NoneType' object has no attribute 'group'")
-    )
+
+    # The exception type depends on the version of importlib.metadata or
+    # its backport, importlib_metadata.
+    # Older versions (<4.11.0) raise AttributeError, newer raise ValueError.
+    # See https://github.com/python/importlib_metadata/pull/384
+    actual_exception = result["foo$bar"]
+    assert isinstance(actual_exception, (AttributeError, ValueError))
 
     _, stderr = capsys.readouterr()
-    assert (
-        "Warning! Failed to load extra entrypoint 'foo$bar': "
-        "AttributeError: 'NoneType' object has no attribute 'group'" in stderr
-    )
+    assert "extra entrypoint" in stderr
+    assert f"{type(actual_exception).__name__}:" in stderr
 
 
 @pytest.mark.usefixtures("module_fixture")
@@ -139,7 +143,9 @@ def test_load_extra_entrypoints_callable_given_no_arguments():
 
 
 @pytest.mark.usefixtures("module_fixture")
-def test_load_extra_entrypoints_callable_that_raises(capsys):
+def test_load_extra_entrypoints_callable_that_raises(
+    capsys: pytest.CaptureFixture[str],
+):
     with temporary_settings(
         {PREFECT_EXTRA_ENTRYPOINTS: "test_module_name:raises_value_error"}
     ):
@@ -167,7 +173,7 @@ def test_load_extra_entrypoints_callable_that_raises_base_exception():
 
 
 @pytest.mark.usefixtures("raising_module_fixture")
-def test_load_extra_entrypoints_error_on_import(capsys):
+def test_load_extra_entrypoints_error_on_import(capsys: pytest.CaptureFixture[str]):
     with temporary_settings({PREFECT_EXTRA_ENTRYPOINTS: "raising_module_name"}):
         result = load_extra_entrypoints()
 
@@ -182,7 +188,7 @@ def test_load_extra_entrypoints_error_on_import(capsys):
 
 
 @pytest.mark.usefixtures("module_fixture")
-def test_load_extra_entrypoints_missing_module(capsys):
+def test_load_extra_entrypoints_missing_module(capsys: pytest.CaptureFixture[str]):
     with temporary_settings({PREFECT_EXTRA_ENTRYPOINTS: "nonexistant_module"}):
         result = load_extra_entrypoints()
 
@@ -200,7 +206,7 @@ def test_load_extra_entrypoints_missing_module(capsys):
 
 
 @pytest.mark.usefixtures("module_fixture")
-def test_load_extra_entrypoints_missing_submodule(capsys):
+def test_load_extra_entrypoints_missing_submodule(capsys: pytest.CaptureFixture[str]):
     with temporary_settings(
         {PREFECT_EXTRA_ENTRYPOINTS: "test_module_name.missing_module"}
     ):
@@ -223,7 +229,7 @@ def test_load_extra_entrypoints_missing_submodule(capsys):
 
 
 @pytest.mark.usefixtures("module_fixture")
-def test_load_extra_entrypoints_missing_attribute(capsys):
+def test_load_extra_entrypoints_missing_attribute(capsys: pytest.CaptureFixture[str]):
     with temporary_settings(
         {PREFECT_EXTRA_ENTRYPOINTS: "test_module_name:missing_attr"}
     ):
@@ -242,7 +248,7 @@ def test_load_extra_entrypoints_missing_attribute(capsys):
     )
 
 
-def test_plugin_load_on_prefect_package_init(monkeypatch):
+def test_plugin_load_on_prefect_package_init(monkeypatch: pytest.MonkeyPatch):
     mock_load_prefect_collections = Mock()
     mock_load_extra_entrypoints = Mock()