Merge branch 'main' into ogenstad-type-hints

nornir-automation · Oct 3, 2023 · 071d077 · 071d077
2 parents 4f66776 + d8c547c
commit 071d077
Show file tree

Hide file tree

Showing 6 changed files with 523 additions and 364 deletions.
diff --git a/nornir/core/plugins/connections.py b/nornir/core/plugins/connections.py
@@ -1,10 +1,8 @@
-from typing import Any, Dict, Type, Optional
+from typing import Any, Dict, Type, Optional, Protocol
 
 from nornir.core.configuration import Config
 from nornir.core.plugins.register import PluginRegister
 
-from typing_extensions import Protocol
-
 
 CONNECTIONS_PLUGIN_PATH = "nornir.plugins.connections"
 

diff --git a/nornir/core/plugins/inventory.py b/nornir/core/plugins/inventory.py
@@ -1,10 +1,8 @@
-from typing import Any, Type
+from typing import Any, Type, Protocol
 
 from nornir.core.inventory import Inventory, TransformFunction
 from nornir.core.plugins.register import PluginRegister
 
-from typing_extensions import Protocol
-
 
 INVENTORY_PLUGIN_PATH = "nornir.plugins.inventory"
 TRANSFORM_FUNCTION_PLUGIN_PATH = "nornir.plugins.transform_function"

diff --git a/nornir/core/plugins/runners.py b/nornir/core/plugins/runners.py
@@ -1,11 +1,9 @@
-from typing import Any, List, Type
+from typing import Any, List, Type, Protocol
 
 from nornir.core.task import AggregatedResult, Task
 from nornir.core.inventory import Host
 from nornir.core.plugins.register import PluginRegister
 
-from typing_extensions import Protocol
-
 
 RUNNERS_PLUGIN_PATH = "nornir.plugins.runners"
 

diff --git a/nornir/core/processor.py b/nornir/core/processor.py
@@ -1,10 +1,8 @@
-from typing import List
+from typing import List, Protocol
 
 from nornir.core.inventory import Host
 from nornir.core.task import AggregatedResult, MultiResult, Task
 
-from typing_extensions import Protocol
-
 
 class Processor(Protocol):
     """