From 2f469b53b64097b8b967cd3b48b4606d126132aa Mon Sep 17 00:00:00 2001 From: Michael Adkins Date: Fri, 6 Jan 2023 11:05:41 -0600 Subject: [PATCH] Extend `visit_collection` with support for annotations (#7263) Co-authored-by: Andrew <15331990+ahuang11@users.noreply.github.com> --- src/prefect/utilities/annotations.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/prefect/utilities/annotations.py b/src/prefect/utilities/annotations.py index 238b72c8f827..0f2ff3054ee8 100644 --- a/src/prefect/utilities/annotations.py +++ b/src/prefect/utilities/annotations.py @@ -37,6 +37,9 @@ def rewrap(self, value: T) -> "BaseAnnotation[T]": def rewrap(self, value: T) -> "BaseAnnotation[T]": return type(self)(value) + def rewrap(self, value: T) -> "BaseAnnotation[T]": + return type(self)(value) + def __eq__(self, other: object) -> bool: if not type(self) == type(other): return False