remove mypy_extensions dep

nornir-automation · Sep 22, 2023 · 1f69e49 · 1f69e49
1 parent d8c547c
commit 1f69e49
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 185 deletions.
diff --git a/nornir/core/inventory.py b/nornir/core/inventory.py
@@ -11,6 +11,7 @@
     ItemsView,
     Iterator,
     TypeVar,
+    Protocol,
 )
 
 from nornir.core.configuration import Config
@@ -20,8 +21,6 @@
 )
 from nornir.core.exceptions import ConnectionAlreadyOpen, ConnectionNotOpen
 
-from mypy_extensions import Arg, KwArg
-
 
 HostOrGroup = TypeVar("HostOrGroup", "Host", "Group")
 
@@ -584,8 +583,14 @@ class Groups(Dict[str, Group]):
     pass
 
 
-TransformFunction = Callable[[Arg(Host), KwArg(Any)], None]
-FilterObj = Callable[[Arg(Host), KwArg(Any)], bool]
+class TransformFunction(Protocol):
+    def __call__(self, host: Host, **kwargs: Any) -> None:
+        ...
+
+
+class FilterObj(Protocol):
+    def __call__(self, host: Host, **kwargs: Any) -> bool:
+        ...
 
 
 class Inventory(object):