Skip to content

Commit d8fe46d

Browse files
committed
fix: merge type checks
-e Signed-off-by: machichima <nary12321@gmail.com>
1 parent b382b14 commit d8fe46d

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

flytekit/interaction/click_types.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -156,11 +156,7 @@ class FileParamType(click.ParamType):
156156
def convert(
157157
self, value: typing.Any, param: typing.Optional[click.Parameter], ctx: typing.Optional[click.Context]
158158
) -> typing.Any:
159-
if isinstance(value, ArtifactQuery):
160-
return value
161-
162-
# If value is already a FlyteFile, return it as-is to avoid creating nested FlyteFile objects
163-
if isinstance(value, FlyteFile):
159+
if isinstance(value, (ArtifactQuery, FlyteFile)):
164160
return value
165161

166162
# set remote_directory to false if running pyflyte run locally. This makes sure that the original

0 commit comments

Comments
 (0)