remove mypy_extensions dep (nornir-automation#870)

jvanderaa · Oct 3, 2023 · 33eb0d3 · 33eb0d3
1 parent 0424c07
commit 33eb0d3
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 186 deletions.
diff --git a/nornir/core/inventory.py b/nornir/core/inventory.py
@@ -1,6 +1,5 @@
 from typing import (
     Any,
-    Callable,
     Dict,
     List,
     Optional,
@@ -11,6 +10,7 @@
     ItemsView,
     Iterator,
     TypeVar,
+    Protocol,
 )
 
 from nornir.core.configuration import Config
@@ -20,8 +20,6 @@
 )
 from nornir.core.exceptions import ConnectionAlreadyOpen, ConnectionNotOpen
 
-from mypy_extensions import Arg, KwArg
-
 
 HostOrGroup = TypeVar("HostOrGroup", "Host", "Group")
 
@@ -584,8 +582,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):