flyte 2.0.0b21__py3-none-any.whl → 2.0.0b22__py3-none-any.whl

flyte/_bin/runtime.py

@@ -137,7 +137,7 @@ def main(
         logger.debug(f"Using controller endpoint: {ep} with kwargs: {controller_kwargs}")
 
     bundle = CodeBundle(tgz=tgz, pkl=pkl, destination=dest, computed_version=version)
-    init(org=org, project=project, domain=domain, **controller_kwargs)
+    init(org=org, project=project, domain=domain, image_builder="remote", **controller_kwargs)
     # Controller is created with the same kwargs as init, so that it can be used to run tasks
     controller = create_controller(ct="remote", **controller_kwargs)
 

flyte/_deploy.py

@@ -1,10 +1,13 @@
 from __future__ import annotations
 
 import asyncio
+import hashlib
+import sys
 import typing
 from dataclasses import dataclass
 from typing import TYPE_CHECKING, Dict, List, Optional, Tuple
 
+import cloudpickle
 import rich.repr
 
 import flyte.errors
@@ -160,9 +163,11 @@ async def _build_images(deployment: DeploymentPlan) -> ImageCache:
         logger.warning(f"Built Image for environment {env_name}, image: {image_uri}")
         env = deployment.envs[env_name]
         if isinstance(env.image, Image):
-            image_identifier_map[env.image.identifier] = image_uri
+            py_version = "{}.{}".format(*env.image.python_version)
+            image_identifier_map[env.image.identifier] = {py_version: image_uri}
         elif env.image == "auto":
-            image_identifier_map["auto"] = image_uri
+            py_version = "{}.{}".format(sys.version_info.major, sys.version_info.minor)
+            image_identifier_map["auto"] = {py_version: image_uri}
 
     return ImageCache(image_lookup=image_identifier_map)
 
@@ -175,15 +180,18 @@ async def apply(deployment_plan: DeploymentPlan, copy_style: CopyFiles, dryrun:
 
     image_cache = await _build_images(deployment_plan)
 
-    version = deployment_plan.version
-    if copy_style == "none" and not version:
+    if copy_style == "none" and not deployment_plan.version:
         raise flyte.errors.DeploymentError("Version must be set when copy_style is none")
     else:
         code_bundle = await build_code_bundle(from_dir=cfg.root_dir, dryrun=dryrun, copy_style=copy_style)
-        version = version or code_bundle.computed_version
-        # TODO we should update the version to include the image cache digest and code bundle digest. This is
-        # to ensure that changes in image dependencies, cause an update to the deployment version.
-        # TODO Also hash the environment and tasks to ensure that changes in the environment or tasks
+        if deployment_plan.version:
+            version = deployment_plan.version
+        else:
+            h = hashlib.md5()
+            h.update(cloudpickle.dumps(deployment_plan.envs))
+            h.update(code_bundle.computed_version.encode("utf-8"))
+            h.update(cloudpickle.dumps(image_cache))
+            version = h.hexdigest()
 
     sc = SerializationContext(
         project=cfg.project,

flyte/_image.py

@@ -203,6 +203,7 @@ class UVProject(PipOption, Layer):
 
         super().update_hash(hasher)
         filehash_update(self.uvlock, hasher)
+        filehash_update(self.pyproject, hasher)
 
 
 @rich.repr.auto
@@ -435,7 +436,10 @@ class Image:
         # Only get the non-None values in the Image to ensure the hash is consistent
         # across different SDK versions.
         # Layers can specify a _compute_identifier optionally, but the default will just stringify
-        image_dict = asdict(self, dict_factory=lambda x: {k: v for (k, v) in x if v is not None and k != "_layers"})
+        image_dict = asdict(
+            self,
+            dict_factory=lambda x: {k: v for (k, v) in x if v is not None and k not in ("_layers", "python_version")},
+        )
         layers_str_repr = "".join([layer.identifier() for layer in self._layers])
         image_dict["layers"] = layers_str_repr
         spec_bytes = image_dict.__str__().encode("utf-8")

flyte/_initialize.py

@@ -33,6 +33,7 @@ class CommonInit:
     project: str | None = None
     domain: str | None = None
     batch_size: int = 1000
+    source_config_path: Optional[Path] = None  # Only used for documentation
 
 
 @dataclass(init=True, kw_only=True, repr=True, eq=True, frozen=True)
@@ -140,6 +141,7 @@ async def init(
     storage: Storage | None = None,
     batch_size: int = 1000,
     image_builder: ImageBuildEngine.ImageBuilderType = "local",
+    source_config_path: Optional[Path] = None,
 ) -> None:
     """
     Initialize the Flyte system with the given configuration. This method should be called before any other Flyte
@@ -175,7 +177,7 @@ async def init(
     :param batch_size: Optional batch size for operations that use listings, defaults to 1000, so limit larger than
       batch_size will be split into multiple requests.
     :param image_builder: Optional image builder configuration, if not provided, the default image builder will be used.
-
+    :param source_config_path: Optional path to the source configuration file (This is only used for documentation)
     :return: None
     """
     from flyte._utils import get_cwd_editable_install, org_from_endpoint, sanitize_endpoint
@@ -224,6 +226,7 @@ async def init(
             org=org or org_from_endpoint(endpoint),
             batch_size=batch_size,
             image_builder=image_builder,
+            source_config_path=source_config_path,
         )
 
 
@@ -245,6 +248,7 @@ async def init_from_config(
         if not available, the current working directory.
     :param log_level: Optional logging level for the framework logger,
         default is set using the default initialization policies
+    :param storage: Optional blob store (S3, GCS, Azure) configuration if needed to access (i.e. using Minio)
     :return: None
     """
     from rich.highlighter import ReprHighlighter
@@ -252,6 +256,7 @@ async def init_from_config(
     import flyte.config as config
 
     cfg: config.Config
+    cfg_path: Optional[Path] = None
     if path_or_config is None:
         # If no path is provided, use the default config file
         cfg = config.auto()
@@ -290,6 +295,7 @@ async def init_from_config(
         log_level=log_level,
         image_builder=cfg.image.builder,
         storage=storage,
+        source_config_path=cfg_path,
     )
 
 

flyte/_interface.py

@@ -7,6 +7,8 @@ from typing import Dict, Generator, Literal, Tuple, Type, TypeVar, Union, cast,
 
 from flyte._logging import logger
 
+LITERAL_ENUM = "LiteralEnum"
+
 
 def default_output_name(index: int = 0) -> str:
     return f"o{index}"
@@ -110,6 +112,6 @@ def literal_to_enum(literal_type: Type) -> Type[Enum | typing.Any]:
     enum_dict = {str(v).upper(): v for v in values}
 
     # Dynamically create an Enum
-    literal_enum = Enum("LiteralEnum", enum_dict)  # type: ignore
+    literal_enum = Enum(LITERAL_ENUM, enum_dict)  # type: ignore
 
     return literal_enum  # type: ignore

flyte/_internal/imagebuild/docker_builder.py

@@ -245,6 +245,9 @@ class UVProjectHandler:
         else:
             # Copy the entire project.
             pyproject_dst = copy_files_to_context(layer.pyproject.parent, context_path)
+            if layer.uvlock:
+                # Sometimes the uv.lock file is in a different folder, if it's specified, let's copy it there explicitly
+                shutil.copy(layer.uvlock, pyproject_dst)
             delta = UV_LOCK_INSTALL_TEMPLATE.substitute(
                 PYPROJECT_PATH=pyproject_dst.relative_to(context_path),
                 PIP_INSTALL_ARGS=" ".join(layer.get_pip_install_args()),

flyte/_internal/imagebuild/image_builder.py

@@ -235,7 +235,7 @@ class ImageBuildEngine:
 
 
 class ImageCache(BaseModel):
-    image_lookup: Dict[str, str]
+    image_lookup: Dict[str, Dict[str, str]]
     serialized_form: str | None = None
 
     @property
@@ -273,10 +273,11 @@ class ImageCache(BaseModel):
         """
         tuples = []
         for k, v in self.image_lookup.items():
-            tuples.append(
-                [
-                    ("Name", k),
-                    ("image", v),
-                ]
-            )
+            for py_version, image_uri in v.items():
+                tuples.append(
+                    [
+                        ("Name", f"{k} (py{py_version})"),
+                        ("image", image_uri),
+                    ]
+                )
         return tuples

flyte/_internal/runtime/task_serde.py

@@ -4,6 +4,7 @@ It includes a Resolver interface for loading tasks, and functions to load classe
 """
 
 import copy
+import sys
 import typing
 from datetime import timedelta
 from typing import Optional, cast
@@ -217,16 +218,25 @@ def _get_urun_container(
         img_uri = task_template.image.uri
     elif serialize_context.image_cache and image_id not in serialize_context.image_cache.image_lookup:
         img_uri = task_template.image.uri
-        from flyte._version import __version__
 
         logger.warning(
-            f"Image {task_template.image} not found in the image cache: {serialize_context.image_cache.image_lookup}.\n"
-            f"This typically occurs when the Flyte SDK version (`{__version__}`) used in the task environment "
-            f"differs from the version used to compile or deploy it.\n"
-            f"Ensure both environments use the same Flyte SDK version to avoid inconsistencies in image resolution."
+            f"Image {task_template.image} not found in the image cache: {serialize_context.image_cache.image_lookup}."
         )
     else:
-        img_uri = serialize_context.image_cache.image_lookup[image_id]
+        python_version_str = "{}.{}".format(sys.version_info.major, sys.version_info.minor)
+        version_lookup = serialize_context.image_cache.image_lookup[image_id]
+        if python_version_str in version_lookup:
+            img_uri = version_lookup[python_version_str]
+        elif version_lookup:
+            # Fallback: try to get any available version
+            fallback_py_version, img_uri = next(iter(version_lookup.items()))
+            logger.warning(
+                f"Image {task_template.image} for python version {python_version_str} "
+                f"not found in the image cache: {serialize_context.image_cache.image_lookup}.\n"
+                f"Fall back using image {img_uri} for python version {fallback_py_version} ."
+            )
+        else:
+            img_uri = task_template.image.uri
 
     return tasks_pb2.Container(
         image=img_uri,

flyte/_keyring/file.py

@@ -72,9 +72,9 @@ class SimplePlainTextKeyring(KeyringBackend):
 
     @property
     def file_path(self) -> Path:
-        from flyte.config._reader import resolve_config_path
+        from flyte._initialize import get_common_config
 
-        config_path = resolve_config_path()
+        config_path = get_common_config().source_config_path
         if config_path and str(config_path.parent) == ".flyte":
             # if the config is in a .flyte directory, use that as the path
             return config_path.parent / "keyring.cfg"

flyte/_task_environment.py

@@ -135,7 +135,7 @@ class TaskEnvironment(Environment):
 
     def task(
         self,
-        _func=None,
+        _func: Callable[P, R] | None = None,
         *,
         short_name: Optional[str] = None,
         cache: CacheRequest | None = None,

flyte/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '2.0.0b21'
-__version_tuple__ = version_tuple = (2, 0, 0, 'b21')
+__version__ = version = '2.0.0b22'
+__version_tuple__ = version_tuple = (2, 0, 0, 'b22')
 
-__commit_id__ = commit_id = 'g7f82f6ddd'
+__commit_id__ = commit_id = 'gce9a6bede'

flyte/cli/_common.py

@@ -111,7 +111,7 @@ class CLIConfig:
         """
         return replace(self, **kwargs)
 
-    def init(self, project: str | None = None, domain: str | None = None):
+    def init(self, project: str | None = None, domain: str | None = None, root_dir: str | None = None):
         from flyte.config._config import TaskConfig
 
         task_cfg = TaskConfig(
@@ -131,7 +131,7 @@ class CLIConfig:
 
         updated_config = self.config.with_params(platform_cfg, task_cfg)
 
-        flyte.init_from_config(updated_config, log_level=self.log_level)
+        flyte.init_from_config(updated_config, log_level=self.log_level, root_dir=root_dir)
 
 
 class InvokeBaseMixin:

flyte/cli/_deploy.py

@@ -43,6 +43,16 @@ class DeployArguments:
             )
         },
     )
+    root_dir: str | None = field(
+        default=None,
+        metadata={
+            "click.option": click.Option(
+                ["--root-dir"],
+                type=str,
+                help="Override the root source directory, helpful when working with monorepos.",
+            )
+        },
+    )
     recursive: bool = field(
         default=False,
         metadata={
@@ -99,7 +109,7 @@ class DeployEnvCommand(click.RichCommand):
         console = Console()
         console.print(f"Deploying root - environment: {self.env_name}")
         obj: CLIConfig = ctx.obj
-        obj.init(self.deploy_args.project, self.deploy_args.domain)
+        obj.init(self.deploy_args.project, self.deploy_args.domain, root_dir=self.deploy_args.root_dir)
         with console.status("Deploying...", spinner="dots"):
             deployment = flyte.deploy(
                 self.env,

flyte/cli/_run.py

@@ -23,14 +23,14 @@ RUN_REMOTE_CMD = "deployed-task"
 
 
 @lru_cache()
-def _initialize_config(ctx: click.Context, project: str, domain: str):
+def _initialize_config(ctx: click.Context, project: str, domain: str, root_dir: str | None = None):
     obj: CLIConfig | None = ctx.obj
     if obj is None:
         import flyte.config
 
         obj = CLIConfig(flyte.config.auto(), ctx)
 
-    obj.init(project, domain)
+    obj.init(project, domain, root_dir)
     return obj
 
 
@@ -77,6 +77,16 @@ class RunArguments:
             )
         },
     )
+    root_dir: str | None = field(
+        default=None,
+        metadata={
+            "click.option": click.Option(
+                ["--root-dir"],
+                type=str,
+                help="Override the root source directory, helpful when working with monorepos.",
+            )
+        },
+    )
     name: str | None = field(
         default=None,
         metadata={
@@ -121,7 +131,7 @@ class RunTaskCommand(click.RichCommand):
         super().__init__(obj_name, *args, **kwargs)
 
     def invoke(self, ctx: click.Context):
-        obj: CLIConfig = _initialize_config(ctx, self.run_args.project, self.run_args.domain)
+        obj: CLIConfig = _initialize_config(ctx, self.run_args.project, self.run_args.domain, self.run_args.root_dir)
 
         async def _run():
             import flyte

flyte/remote/_action.py

@@ -369,14 +369,15 @@ class Action(ToJSONMixin):
                     # If the action is done, handle the final state
                     if ad.done():
                         progress.stop_task(task_id)
-                        if ad.pb2.status.phase == run_definition_pb2.PHASE_SUCCEEDED:
-                            console.print(f"[bold green]Run '{self.run_name}' completed successfully.[/bold green]")
-                        else:
-                            error_message = ad.error_info.message if ad.error_info else ""
-                            console.print(
-                                f"[bold red]Run '{self.run_name}' exited unsuccessfully in state {ad.phase}"
-                                f" with error: {error_message}[/bold red]"
-                            )
+                        if not quiet:
+                            if ad.pb2.status.phase == run_definition_pb2.PHASE_SUCCEEDED:
+                                console.print(f"[bold green]Run '{self.run_name}' completed successfully.[/bold green]")
+                            else:
+                                error_message = ad.error_info.message if ad.error_info else ""
+                                console.print(
+                                    f"[bold red]Run '{self.run_name}' exited unsuccessfully in state {ad.phase}"
+                                    f" with error: {error_message}[/bold red]"
+                                )
                         break
         except asyncio.CancelledError:
             # Handle cancellation gracefully

flyte/types/_pickle.py

@@ -1,5 +1,6 @@
 import hashlib
 import os
+import sys
 import typing
 from typing import Type
 
@@ -12,6 +13,7 @@ import flyte.storage as storage
 from ._type_engine import TypeEngine, TypeTransformer
 
 T = typing.TypeVar("T")
+DEFAULT_PICKLE_BYTES_LIMIT = 2**10 * 10  # 10KB
 
 
 class FlytePickle(typing.Generic[T]):
@@ -76,8 +78,13 @@ class FlytePickleTransformer(TypeTransformer[FlytePickle]):
         ...
 
     async def to_python_value(self, lv: literals_pb2.Literal, expected_python_type: Type[T]) -> T:
-        uri = lv.scalar.blob.uri
-        return await FlytePickle.from_pickle(uri)
+        if lv.scalar.blob and lv.scalar.blob.uri:
+            uri = lv.scalar.blob.uri
+            return await FlytePickle.from_pickle(uri)
+        elif lv.scalar.binary and lv.scalar.binary.value:
+            return cloudpickle.loads(lv.scalar.binary.value)
+        else:
+            raise ValueError(f"Cannot convert {lv} to {expected_python_type}")
 
     async def to_literal(
         self,
@@ -92,8 +99,15 @@ class FlytePickleTransformer(TypeTransformer[FlytePickle]):
                 format=self.PYTHON_PICKLE_FORMAT, dimensionality=types_pb2.BlobType.BlobDimensionality.SINGLE
             )
         )
-        remote_path = await FlytePickle.to_pickle(python_val)
-        return literals_pb2.Literal(scalar=literals_pb2.Scalar(blob=literals_pb2.Blob(metadata=meta, uri=remote_path)))
+        if sys.getsizeof(python_val) > DEFAULT_PICKLE_BYTES_LIMIT:
+            remote_path = await FlytePickle.to_pickle(python_val)
+            return literals_pb2.Literal(
+                scalar=literals_pb2.Scalar(blob=literals_pb2.Blob(metadata=meta, uri=remote_path))
+            )
+        else:
+            return literals_pb2.Literal(
+                scalar=literals_pb2.Scalar(binary=literals_pb2.Binary(value=cloudpickle.dumps(python_val)))
+            )
 
     def guess_python_type(self, literal_type: types_pb2.LiteralType) -> typing.Type[FlytePickle[typing.Any]]:
         if (

flyte/types/_type_engine.py

@@ -803,6 +803,12 @@ class EnumTransformer(TypeTransformer[enum.Enum]):
     async def to_python_value(self, lv: Literal, expected_python_type: Type[T]) -> T:
         if lv.HasField("scalar") and lv.scalar.HasField("binary"):
             return self.from_binary_idl(lv.scalar.binary, expected_python_type)  # type: ignore
+        from flyte._interface import LITERAL_ENUM
+
+        if expected_python_type.__name__ is LITERAL_ENUM:
+            # This is the case when python Literal types are used as enums. The class name is always LiteralEnum an
+            # hardcoded in flyte.models
+            return lv.scalar.primitive.string_value
         return expected_python_type(lv.scalar.primitive.string_value)  # type: ignore
 
     def guess_python_type(self, literal_type: LiteralType) -> Type[enum.Enum]:

{flyte-2.0.0b21.data → flyte-2.0.0b22.data}/scripts/runtime.py

@@ -137,7 +137,7 @@ def main(
         logger.debug(f"Using controller endpoint: {ep} with kwargs: {controller_kwargs}")
 
     bundle = CodeBundle(tgz=tgz, pkl=pkl, destination=dest, computed_version=version)
-    init(org=org, project=project, domain=domain, **controller_kwargs)
+    init(org=org, project=project, domain=domain, image_builder="remote", **controller_kwargs)
     # Controller is created with the same kwargs as init, so that it can be used to run tasks
     controller = create_controller(ct="remote", **controller_kwargs)
 

{flyte-2.0.0b21.dist-info → flyte-2.0.0b22.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flyte
-Version: 2.0.0b21
+Version: 2.0.0b22
 Summary: Add your description here
 Author-email: Ketan Umare <kumare3@users.noreply.github.com>
 Requires-Python: >=3.10

{flyte-2.0.0b21.dist-info → flyte-2.0.0b22.dist-info}/RECORD

@@ -1,16 +1,16 @@
 flyte/__init__.py,sha256=RIvRNnyKc9TfF5FlvREETDuosgs66w6d_A-MPMbyy9E,2158
 flyte/_build.py,sha256=MkgfLAPeL56YeVrGRNZUCZgbwzlEzVP3wLbl5Qru4yk,578
 flyte/_context.py,sha256=NGl-tDoTOoIljsUzD1IPTESOdaG8vKeoAzs5271XuPM,5244
-flyte/_deploy.py,sha256=2V1OpmP2Rb3j0BuxGhjYzUcF--NrmAoLVO0Eq0usnw0,10635
+flyte/_deploy.py,sha256=Y-YjPXtlnFGlP3Yuy6N_DhVBwwcUGPEYWImJfqOWcyE,10816
 flyte/_doc.py,sha256=_OPCf3t_git6UT7kSJISFaWO9cfNzJhhoe6JjVdyCJo,706
 flyte/_docstring.py,sha256=SsG0Ab_YMAwy2ABJlEo3eBKlyC3kwPdnDJ1FIms-ZBQ,1127
 flyte/_environment.py,sha256=XYGscZ_I4xw-52o1ZULCGCVBY-l2m98rp9wpiQO4tQw,4590
 flyte/_excepthook.py,sha256=nXts84rzEg6-7RtFarbKzOsRZTQR4plnbWVIFMAEprs,1310
 flyte/_group.py,sha256=7o1j16sZyUmYB50mOiq1ui4TBAKhRpDqLakV8Ya1kw4,803
 flyte/_hash.py,sha256=KMKjoI7SaxXildb-xv6n5Vb32B0csvBiYc06iUe-BrI,137
-flyte/_image.py,sha256=CSbH7XSSRSNtzri5hLOR-tKIaFW2m8VMEoC1TZmdg6M,38602
-flyte/_initialize.py,sha256=XzbAtqPM83ml4jH14kvClUCeKshchFQUvFZghGZVGQM,17923
-flyte/_interface.py,sha256=LBprgsS_DIx5fdnWUyeKQnl2XlAyc1SYlRNlZySwRIA,4766
+flyte/_image.py,sha256=fMVV7G13BrAm8S1TzubOlmneOeW6novLSGYchcBg6SQ,38709
+flyte/_initialize.py,sha256=nIeZVVO0sQjQh6F9P5rOSq5I2SXumnLwbPYzavdzRhY,18396
+flyte/_interface.py,sha256=krUJOv43dOsSUNGlt30ajfFUl8_HPhubPg6dRlGSa4k,4795
 flyte/_logging.py,sha256=7e_rQC4gtBcKvJStNX8Af-ZJG0U0r-2Om_8JMvkvb5k,6283
 flyte/_map.py,sha256=8u4jZsM26V-M-dhVbSnyqSrkzJnGgisN3kwwSNQZaa4,10697
 flyte/_pod.py,sha256=MB5eP2WvTc5lD5ovdVlxoOpx-wJeutv0S64s75astLc,1105
@@ -20,19 +20,19 @@ flyte/_reusable_environment.py,sha256=qzmLJlHFiek8_k3EEqxew3837Pe2xjmz3mjGk_xqPE
 flyte/_run.py,sha256=N5N2LPFIOgVBflGaL7WAErQyspGa3zlKuEQ1uoqVTuA,25901
 flyte/_secret.py,sha256=wug5NbIYEkXO6FJkqnPRnPoc2nKDerQiemWtRGRi288,3576
 flyte/_task.py,sha256=kYoguvcRXwkwCbFBVCl6MXXYKHgADl2H53qWcs4CEQ4,20201
-flyte/_task_environment.py,sha256=F4ZmitZi845zw-VwTHvC78qpuBEzmeR3cYObgmE8jTQ,10061
+flyte/_task_environment.py,sha256=jwvi6ceYoqwnvEFEJR5KmTZHngsuhAkFfWmhtRcl1rk,10086
 flyte/_task_plugins.py,sha256=9MH3nFPOH_e8_92BT4sFk4oyAnj6GJFvaPYWaraX7yE,1037
 flyte/_timeout.py,sha256=zx5sFcbYmjJAJbZWSGzzX-BpC9HC7Jfs35T7vVhKwkk,1571
 flyte/_tools.py,sha256=lB3OiJSAhxzSMCYjLUF6nZjlFsmNpaRXtr3_Fefcxbg,747
 flyte/_trace.py,sha256=-BIprs2MbupWl3vsC_Pn33SV3fSVku1rUIsnwfmrIy0,5204
-flyte/_version.py,sha256=yhZDd6WrjR7VGM3U8AX0LKHWlEqOxtftv0MsV5z9dR0,722
+flyte/_version.py,sha256=kBIvRTcJoPn-XOZzko2zml9uPZwZAzL5typ9a4VENRE,722
 flyte/errors.py,sha256=k1-pgz7xm4syViJmPwp7ZDusmqcrlM2uZKrl0com_9I,6707
 flyte/extend.py,sha256=GB4ZedGzKa30vYWRVPOdxEeK62xnUVFY4z2tD6H9eEw,376
 flyte/models.py,sha256=l7h-o6pLHGqFcbymAOagHdR4BGKBLJUs6Pmd4J1xbQU,16409
 flyte/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 flyte/_bin/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 flyte/_bin/debug.py,sha256=hnX2tlv9QbqckoT5CJ3c3apJj3tGDpsrdV7ZAsE7j34,911
-flyte/_bin/runtime.py,sha256=SWSHMHGPQdKSjbc9yJkBbLOXUIpPq8iBk0eFsKDdCeM,6203
+flyte/_bin/runtime.py,sha256=4MUSHHPznY6mcfy9nEr48KaM2pFIIScEei_J779r6KU,6227
 flyte/_cache/__init__.py,sha256=zhdO5UuHQRdzn8GHmSN40nrxfAmI4ihDRuHZM11U84Y,305
 flyte/_cache/cache.py,sha256=-oS5q6QpkxPqZ8gWkn8uVNgzx3JFuXaWLV-h0KVKW9c,5010
 flyte/_cache/defaults.py,sha256=gzJZW0QJPUfd2OPnGpv3tzIfwPtgFjAKoie3NP1P97U,217
@@ -58,8 +58,8 @@ flyte/_internal/controllers/remote/_core.py,sha256=5qIGFNTilJwtYLQ_KQoawLQ172ih-
 flyte/_internal/controllers/remote/_informer.py,sha256=M2uqfe0oz9ySkYK94UFLoTPRt9UoaYQKzmy4lrVOchY,14912
 flyte/_internal/controllers/remote/_service_protocol.py,sha256=B9qbIg6DiGeac-iSccLmX_AL2xUgX4ezNUOiAbSy4V0,1357
 flyte/_internal/imagebuild/__init__.py,sha256=dwXdJ1jMhw9RF8itF7jkPLanvX1yCviSns7hE5eoIts,102
-flyte/_internal/imagebuild/docker_builder.py,sha256=3PqIh8NGiNo8f9ekUxyZYF4i9I40BD06VoEGaV-ymoo,23840
-flyte/_internal/imagebuild/image_builder.py,sha256=dXBXl62qcPabus6dR3eP8P9mBGNhpZHZ2Xm12AymKkk,11150
+flyte/_internal/imagebuild/docker_builder.py,sha256=jaKs67PqPKTfQaUWN6VHF_TN9jWLSWJgJLw6Ltn9RoA,24047
+flyte/_internal/imagebuild/image_builder.py,sha256=2nqcTX0ry2lxTQYmHZzyXZ9sIcfir6a9fOCNLZkAv6Y,11267
 flyte/_internal/imagebuild/remote_builder.py,sha256=QOs0miveAnDEDxXCNRnkblw7DB-BJwQUOeL0Y_9ao2A,13120
 flyte/_internal/imagebuild/utils.py,sha256=2MMCNyyMoPLO8sJ5J9TnrZwd1xWTty5AW2VVp8ZUd_M,1250
 flyte/_internal/resolvers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -73,11 +73,11 @@ flyte/_internal/runtime/io.py,sha256=ysL7hMpfVumvsEYWOM-_VPa8MXn5_X_CZorKbOThyv4
 flyte/_internal/runtime/resources_serde.py,sha256=TObMVsSjVcQhcY8-nY81pbvrz7TP-adDD5xV-LqAaxM,4813
 flyte/_internal/runtime/reuse.py,sha256=uGjomQz4X4JvPv6M8eDW94g4YfHJugNYufwK6eVFlwI,4901
 flyte/_internal/runtime/rusty.py,sha256=e2uSg9-Hooa65-BIuqXhIrEq86RHteFFs3W7dDuM3uo,6946
-flyte/_internal/runtime/task_serde.py,sha256=lE1x_hEEvPkRN_ogVqWL76dGruG2pZt1q0LyCE2d-f4,14093
+flyte/_internal/runtime/task_serde.py,sha256=ZDym2IujC70KzePrDnB0GteaCcP-87x_C44n3bLmr_Q,14504
 flyte/_internal/runtime/taskrunner.py,sha256=T1dviDXYds18eSwT_J6AMvD4oY91t1r_31EAFA5grO0,7683
 flyte/_internal/runtime/types_serde.py,sha256=EjRh9Yypx9-20XXQprtNgp766LeQVRoYWtY6XPGMZQg,1813
 flyte/_keyring/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-flyte/_keyring/file.py,sha256=6XxDnk_hejWRnH8WLd7RVFqoiRY23Td8Cm2nh1VCloo,2950
+flyte/_keyring/file.py,sha256=MKAHv7B6rFV0gJN509jSMSG7beu-wEpizvDaQ1uuwLY,2962
 flyte/_protos/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 flyte/_protos/common/authorization_pb2.py,sha256=6G7CAfq_Vq1qrm8JFkAnMAj0AaEipiX7MkjA7nk91-M,6707
 flyte/_protos/common/authorization_pb2.pyi,sha256=tdqc3wZo3Yc6lKfjVgJlUFUFzGv4GAaCknIv43RGd-8,4759
@@ -166,15 +166,15 @@ flyte/_utils/uv_script_parser.py,sha256=PxqD8lSMi6xv0uDd1s8LKB2IPZr4ttZJCUweqlyM
 flyte/cli/__init__.py,sha256=aeCcumeP9xD_5aCmaRYUPCe2QRJSGCaxcUbTZ3co768,341
 flyte/cli/_abort.py,sha256=Ty-63Gtd2PUn6lCuL5AaasfBoPu7TDSU5EQKVbkF4qw,661
 flyte/cli/_build.py,sha256=S8I53ou-rt3m54A_Sjg2sPU_Cw7D1GpkS5DO766sfk4,3534
-flyte/cli/_common.py,sha256=9BoRYzQBxuKdczC13u_2auDV8wYABvf3lWZhPcfGclA,13591
+flyte/cli/_common.py,sha256=36VI1dPaFBjzmvqUd-5Pq3VEg5Tp7NAlwtfRFrVYChA,13639
 flyte/cli/_create.py,sha256=3nNf5XxsynITyJljvDYlj2la7ywARF9Tg7GbuIOo1Yw,5640
 flyte/cli/_delete.py,sha256=VTmXv09PBjkdtyl23mbSjIQQlN7Y1AI_bO0GkHP-f9E,546
-flyte/cli/_deploy.py,sha256=mUCq8ZVcl3p511lcf2sY3Z9r2INCzTMKgKoRg2_Su-8,8881
+flyte/cli/_deploy.py,sha256=EhdNaz1sQR1M16-hi2L7mpA0I_SeUgohirFFv4ogNwE,9220
 flyte/cli/_gen.py,sha256=7K2eYQLGVr26I2OC3Xe_bzAn4ANYA5mPlBW5m1476PM,6079
 flyte/cli/_get.py,sha256=E2HTekZcVbTtwy6LnM5WHzF41aT1uPRWqV1rYB4H5B8,10420
 flyte/cli/_option.py,sha256=oC1Gs0u0UrOC1SsrFo-iCuAkqQvI1wJWCdjYXA9rW4Q,1445
 flyte/cli/_params.py,sha256=oCSnBh6fDSS1HLFZvNRstfXlaJHTWaOVwMQ_R6icehs,20009
-flyte/cli/_run.py,sha256=bd024EikZwqHBSLCr1K45bkwwXMD8vxnl_0EIQAh4T0,16376
+flyte/cli/_run.py,sha256=yP8lh5Zgl2y1-tLCH9d7GWSNYJK03oSo1quO00GC_Sk,16742
 flyte/cli/main.py,sha256=t5Ivjipd6bVHIGjRBGwkeP577j59ASq9c1wgoNf3h2c,5334
 flyte/config/__init__.py,sha256=MiwEYK5Iv7MRR22z61nzbsbvZ9Q6MdmAU_g9If1Pmb8,144
 flyte/config/_config.py,sha256=gsGtIZa9CUILfBmWm2CaEMRLHcSOeNUgwPHeIV807bs,10942
@@ -193,7 +193,7 @@ flyte/io/_dataframe/__init__.py,sha256=SDgNw45uf7m3cHhbUCOA3V3-5A2zSKgPcsWriRLwd
 flyte/io/_dataframe/basic_dfs.py,sha256=tUJKiBbIJ30r9hjwMo-y3bxjdLmyf3WVe9n9tYskKWk,8043
 flyte/io/_dataframe/dataframe.py,sha256=HOW6W12uAlN_1YH9fOgk-HaAse3ACgVyb18vsa83Uzo,49371
 flyte/remote/__init__.py,sha256=y9eke9JzEJkygk8eKZjSj44fJGlyepuW4i-j6lbCAPY,617
-flyte/remote/_action.py,sha256=o19TUh7qo5JrCU1NhqXWAPKCWco6yjrvKVP5gt6fskU,24254
+flyte/remote/_action.py,sha256=wYP_tD0tOnjhOAUP3CzyJfiE6tzE2IePXa48qf8Q3qs,24324
 flyte/remote/_common.py,sha256=2XLLxWL1NjwfdPQUhOfYn3zMrg-yGNfi6NYIaqHutUA,862
 flyte/remote/_console.py,sha256=avmELJPx8nQMAVPrHlh6jEIRPjrMwFpdZjJsWOOa9rE,660
 flyte/remote/_data.py,sha256=WeRkhUrJlx7rU3CpvbPvikosv7r02UMveWrf70cFilQ,6156
@@ -235,16 +235,16 @@ flyte/syncify/__init__.py,sha256=WgTk-v-SntULnI55CsVy71cxGJ9Q6pxpTrhbPFuouJ0,197
 flyte/syncify/_api.py,sha256=k4LQB8odJb5Fx2dabL340g0Tq1bKfSG_ZHsV5qRodE0,14858
 flyte/types/__init__.py,sha256=9310PRtVrwJePwEPeoUO0HPyIkgaja7-Dar_QlE_MUI,1745
 flyte/types/_interface.py,sha256=5y9EC5r897xz03Hh0ltF8QVGKMfMfAznws-hKSEO4Go,1677
-flyte/types/_pickle.py,sha256=PjdR66OTDMZ3OYq6GvM_Ua0cIo5t2XQaIjmpJ9xo4Ys,4050
+flyte/types/_pickle.py,sha256=onpJDslzTugnicbqbkBkCFe7JW0YtU06xkp6Yd7ry70,4661
 flyte/types/_renderer.py,sha256=ygcCo5l60lHufyQISFddZfWwLlQ8kJAKxUT_XnR_6dY,4818
 flyte/types/_string_literals.py,sha256=NlG1xV8RSA-sZ-n-IFQCAsdB6jXJOAKkHWtnopxVVDk,4231
-flyte/types/_type_engine.py,sha256=CKWWGwSOP3vFZUlMnv-Rgzu-EJ-DkX--Yhn_g3M9cjw,95394
+flyte/types/_type_engine.py,sha256=xGxuNQK2kShdvxcAvN5v0yP704EXMraM_d7gBPJgHn4,95711
 flyte/types/_utils.py,sha256=pbts9E1_2LTdLygAY0UYTLYJ8AsN3BZyviSXvrtcutc,2626
-flyte-2.0.0b21.data/scripts/debug.py,sha256=hnX2tlv9QbqckoT5CJ3c3apJj3tGDpsrdV7ZAsE7j34,911
-flyte-2.0.0b21.data/scripts/runtime.py,sha256=SWSHMHGPQdKSjbc9yJkBbLOXUIpPq8iBk0eFsKDdCeM,6203
-flyte-2.0.0b21.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-flyte-2.0.0b21.dist-info/METADATA,sha256=P0WPet5aQBjjrj1JvlecH69jE20-AunhqhIGWkMld98,10045
-flyte-2.0.0b21.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-flyte-2.0.0b21.dist-info/entry_points.txt,sha256=rb43Gfxw40iPH5B6EUs6Ter0ekLkGXsj7R890_MOTyk,136
-flyte-2.0.0b21.dist-info/top_level.txt,sha256=7dkyFbikvA12LEZEqawx8oDG1CMod6hTliPj7iWzgYo,6
-flyte-2.0.0b21.dist-info/RECORD,,
+flyte-2.0.0b22.data/scripts/debug.py,sha256=hnX2tlv9QbqckoT5CJ3c3apJj3tGDpsrdV7ZAsE7j34,911
+flyte-2.0.0b22.data/scripts/runtime.py,sha256=4MUSHHPznY6mcfy9nEr48KaM2pFIIScEei_J779r6KU,6227
+flyte-2.0.0b22.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+flyte-2.0.0b22.dist-info/METADATA,sha256=GMmE_XSJ2wEFNqxe5C89PJeznQhVEmP531uc2HIHruk,10045
+flyte-2.0.0b22.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+flyte-2.0.0b22.dist-info/entry_points.txt,sha256=rb43Gfxw40iPH5B6EUs6Ter0ekLkGXsj7R890_MOTyk,136
+flyte-2.0.0b22.dist-info/top_level.txt,sha256=7dkyFbikvA12LEZEqawx8oDG1CMod6hTliPj7iWzgYo,6
+flyte-2.0.0b22.dist-info/RECORD,,