Extend visit_collection with support for annotations (#7263)

Co-authored-by: Andrew <15331990+ahuang11@users.noreply.github.com>
PrefectHQ · Feb 6, 2023 · 995361c · 995361c
1 parent 4cfb889
commit 995361c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/prefect/utilities/annotations.py b/src/prefect/utilities/annotations.py
@@ -40,6 +40,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