diff --git a/flytekit/interaction/click_types.py b/flytekit/interaction/click_types.py index 7918339da3..056fa2db61 100644 --- a/flytekit/interaction/click_types.py +++ b/flytekit/interaction/click_types.py @@ -156,8 +156,9 @@ class FileParamType(click.ParamType): def convert( self, value: typing.Any, param: typing.Optional[click.Parameter], ctx: typing.Optional[click.Context] ) -> typing.Any: - if isinstance(value, ArtifactQuery): + if isinstance(value, (ArtifactQuery, FlyteFile)): return value + # set remote_directory to false if running pyflyte run locally. This makes sure that the original # file is used and not a random one. remote_path = None if getattr(ctx.obj, "is_remote", False) else False @@ -177,6 +178,7 @@ class PickleParamType(click.ParamType): def get_metavar(self, param: Parameter, ctx: Context) -> t.Optional[str]: return "Python Object :" + else: def get_metavar(self, param: Parameter, *args) -> t.Optional[str]: