fix: Make FlyteFile compatible with Annotated[..., HashMethod]

See issue #3424
flyteorg · Mar 9, 2023 · 7cbd2b8 · 7cbd2b8
1 parent be24c52
commit 7cbd2b8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/flytekit/types/file/file.py b/flytekit/types/file/file.py
@@ -272,6 +272,10 @@ def to_literal(
         if python_val is None:
             raise TypeTransformerFailedError("None value cannot be converted to a file.")
 
+        # Correctly handle `Annotated[FlyteFile, ...]` by extracting the origin type
+        if typing.get_origin(python_type) is typing.Annotated:
+            python_type = typing.get_args(python_type)[0]
+
         if not (python_type is os.PathLike or issubclass(python_type, FlyteFile)):
             raise ValueError(f"Incorrect type {python_type}, must be either a FlyteFile or os.PathLike")