From 668f57860a085ca8ecec1ddfb0bfe1a15f3bf9f0 Mon Sep 17 00:00:00 2001 From: Ilmari Aho Date: Fri, 13 Sep 2024 13:02:02 +0300 Subject: [PATCH] fix(#39): correct result typing --- nornir_utils/plugins/functions/print_result.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/nornir_utils/plugins/functions/print_result.py b/nornir_utils/plugins/functions/print_result.py index 01d252d..4de74c5 100644 --- 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,