Extend visit_collection with support for annotations (#7263)

Co-authored-by: Andrew <15331990+ahuang11@users.noreply.github.com>
PrefectHQ · Jan 12, 2023 · 92ccad9 · 92ccad9
1 parent 8e38d51
commit 92ccad9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/prefect/utilities/annotations.py b/src/prefect/utilities/annotations.py
@@ -28,6 +28,9 @@ def unwrap(self) -> 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