fix(#39): correct result typing (#40)

Co-authored-by: David Barroso <[email protected]>
nornir-automation · Sep 18, 2024 · 38f8036 · 38f8036
1 parent 48a2eb4
commit 38f8036
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/nornir_utils/plugins/functions/print_result.py b/nornir_utils/plugins/functions/print_result.py
@@ -1,7 +1,7 @@
 import logging
 import pprint
 import threading
-from typing import List, cast, Optional
+from typing import List, cast, Optional, Union
 from collections import OrderedDict
 import json
 
@@ -77,7 +77,7 @@ def _print_individual_result(
 
 
 def _print_result(
-    result: Result,
+    result: Union[Result, AggregatedResult, MultiResult],
     attrs: Optional[List[str]] = None,
     failed: bool = False,
     severity_level: int = logging.INFO,
@@ -123,7 +123,7 @@ def _print_result(
 
 
 def print_result(
-    result: Result,
+    result: Union[Result, AggregatedResult, MultiResult],
     vars: Optional[List[str]] = None,
     failed: bool = False,
     severity_level: int = logging.INFO,