Use pluggy's new public HookCaller type for type hints

jpmorganchase · Nov 1, 2023 · b6d2656 · b6d2656
1 parent a0c2200
commit b6d2656
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -48,7 +48,7 @@ dependencies = [
     "fastavro~=1.8",  # TODO: consider moving Avro-related dependencies to optional dependencies
     "memoization~=0.4",
     "orjson~=3.0",
-    "pluggy~=1.2",
+    "pluggy~=1.3",
     "py-avro-schema~=3.0",
     "python-dateutil~=2.8",
 ]

diff --git a/src/py_adapter/plugin/__init__.py b/src/py_adapter/plugin/__init__.py
@@ -22,8 +22,6 @@
 import pluggy
 
 if TYPE_CHECKING:
-    from pluggy._hooks import _HookCaller
-
     import py_adapter
 
 logger = logging.getLogger(__package__)
@@ -66,7 +64,7 @@ def _load_default_plugins(manager_: pluggy.PluginManager) -> None:
         manager_.register(plugin, name=name)
 
 
-def plugin_hook(plugin_name: str, hook_name: str) -> "_HookCaller":
+def plugin_hook(plugin_name: str, hook_name: str) -> pluggy.HookCaller:
     """
     Return a hook (caller) for a single named plugin and hook name