flyte 2.0.0b7__py3-none-any.whl → 2.0.0b9__py3-none-any.whl

flyte/_image.py

@@ -154,6 +154,7 @@ class PipPackages(PipOption, Layer):
 class PythonWheels(PipOption, Layer):
     wheel_dir: Path = field(metadata={"identifier": False})
     wheel_dir_name: str = field(init=False)
+    package_name: str
 
     def __post_init__(self):
         object.__setattr__(self, "wheel_dir_name", self.wheel_dir.name)
@@ -498,7 +499,8 @@ class Image:
                     image = image.with_pip_packages(f"flyte=={flyte_version}", pre=True)
                 else:
                     image = image.with_pip_packages(f"flyte=={flyte_version}")
-        object.__setattr__(image, "_tag", preset_tag)
+        if not dev_mode:
+            object.__setattr__(image, "_tag", preset_tag)
         # Set this to auto for all auto images because the meaning of "auto" can change (based on logic inside
         # _get_default_image_for, acts differently in a running task container) so let's make sure it stays auto.
         object.__setattr__(image, "_identifier_override", "auto")
@@ -546,7 +548,7 @@ class Image:
             platform=platform,
         )
 
-        if registry and name:
+        if registry or name:
             return base_image.clone(registry=registry, name=name)
 
         # # Set this to auto for all auto images because the meaning of "auto" can change (based on logic inside
@@ -940,7 +942,7 @@ class Image:
         dist_folder = Path(__file__).parent.parent.parent / "dist"
         # Manually declare the PythonWheel so we can set the hashing
         # used to compute the identifier. Can remove if we ever decide to expose the lambda in with_ commands
-        with_dist = self.clone(addl_layer=PythonWheels(wheel_dir=dist_folder))
+        with_dist = self.clone(addl_layer=PythonWheels(wheel_dir=dist_folder, package_name="flyte", pre=True))
 
         return with_dist
 

flyte/_internal/imagebuild/docker_builder.py

@@ -37,6 +37,7 @@ from flyte._internal.imagebuild.image_builder import (
     LocalDockerCommandImageChecker,
     LocalPodmanCommandImageChecker,
 )
+from flyte._internal.imagebuild.utils import copy_files_to_context
 from flyte._logging import logger
 
 _F_IMG_ID = "_F_IMG_ID"
@@ -60,7 +61,7 @@ ENV PATH="/root/.venv/bin:$$PATH" \
 
 UV_PACKAGE_INSTALL_COMMAND_TEMPLATE = Template("""\
 RUN --mount=type=cache,sharing=locked,mode=0777,target=/root/.cache/uv,id=uv \
-    --mount=type=bind,target=requirements_uv.txt,src=requirements_uv.txt \
+    $REQUIREMENTS_MOUNT \
     $SECRET_MOUNT \
     uv pip install --python $$UV_PYTHON $PIP_INSTALL_ARGS
 """)
@@ -109,7 +110,7 @@ RUN uv venv $$VIRTUALENV --python=$PYTHON_VERSION
 
 # Adds nvidia just in case it exists
 ENV PATH="$$PATH:/usr/local/nvidia/bin:/usr/local/cuda/bin" \
-    LD_LIBRARY_PATH="/usr/local/nvidia/lib64:$$LD_LIBRARY_PATH"
+    LD_LIBRARY_PATH="/usr/local/nvidia/lib64"
 """)
 
 # This gets added on to the end of the dockerfile
@@ -128,30 +129,34 @@ class Handler(Protocol):
 class PipAndRequirementsHandler:
     @staticmethod
     async def handle(layer: PipPackages, context_path: Path, dockerfile: str) -> str:
+        secret_mounts = _get_secret_mounts_layer(layer.secret_mounts)
+
+        # Set pip_install_args based on the layer type - either a requirements file or a list of packages
         if isinstance(layer, Requirements):
             if not layer.file.exists():
                 raise FileNotFoundError(f"Requirements file {layer.file} does not exist")
             if not layer.file.is_file():
                 raise ValueError(f"Requirements file {layer.file} is not a file")
 
-            async with aiofiles.open(layer.file) as f:
-                requirements = []
-                async for line in f:
-                    requirement = line
-                    requirements.append(requirement.strip())
+            # Copy the requirements file to the context path
+            requirements_path = copy_files_to_context(layer.file, context_path)
+            rel_path = str(requirements_path.relative_to(context_path))
+            pip_install_args = layer.get_pip_install_args()
+            pip_install_args.extend(["--requirement", "requirements.txt"])
+            mount = f"--mount=type=bind,target=requirements.txt,src={rel_path}"
         else:
+            mount = ""
             requirements = list(layer.packages) if layer.packages else []
-        requirements_uv_path = context_path / "requirements_uv.txt"
-        async with aiofiles.open(requirements_uv_path, "w") as f:
-            reqs = "\n".join(requirements)
-            await f.write(reqs)
+            reqs = " ".join(requirements)
+            pip_install_args = layer.get_pip_install_args()
+            pip_install_args.append(reqs)
 
-        pip_install_args = layer.get_pip_install_args()
-        pip_install_args.extend(["--requirement", "requirements_uv.txt"])
-        secret_mounts = _get_secret_mounts_layer(layer.secret_mounts)
         delta = UV_PACKAGE_INSTALL_COMMAND_TEMPLATE.substitute(
-            PIP_INSTALL_ARGS=" ".join(pip_install_args), SECRET_MOUNT=secret_mounts
+            SECRET_MOUNT=secret_mounts,
+            REQUIREMENTS_MOUNT=mount,
+            PIP_INSTALL_ARGS=" ".join(pip_install_args),
         )
+
         dockerfile += delta
 
         return dockerfile
@@ -162,12 +167,31 @@ class PythonWheelHandler:
     async def handle(layer: PythonWheels, context_path: Path, dockerfile: str) -> str:
         shutil.copytree(layer.wheel_dir, context_path / "dist", dirs_exist_ok=True)
         pip_install_args = layer.get_pip_install_args()
-        pip_install_args.extend(["/dist/*.whl"])
         secret_mounts = _get_secret_mounts_layer(layer.secret_mounts)
-        delta = UV_WHEEL_INSTALL_COMMAND_TEMPLATE.substitute(
-            PIP_INSTALL_ARGS=" ".join(pip_install_args), SECRET_MOUNT=secret_mounts
+
+        # First install: Install the wheel without dependencies using --no-deps
+        pip_install_args_no_deps = [
+            *pip_install_args,
+            *[
+                "--find-links",
+                "/dist",
+                "--no-deps",
+                "--no-index",
+                layer.package_name,
+            ],
+        ]
+
+        delta1 = UV_WHEEL_INSTALL_COMMAND_TEMPLATE.substitute(
+            PIP_INSTALL_ARGS=" ".join(pip_install_args_no_deps), SECRET_MOUNT=secret_mounts
         )
-        dockerfile += delta
+        dockerfile += delta1
+
+        # Second install: Install dependencies from PyPI
+        pip_install_args_deps = [*pip_install_args, layer.package_name]
+        delta2 = UV_WHEEL_INSTALL_COMMAND_TEMPLATE.substitute(
+            PIP_INSTALL_ARGS=" ".join(pip_install_args_deps), SECRET_MOUNT=secret_mounts
+        )
+        dockerfile += delta2
 
         return dockerfile
 

flyte/_internal/imagebuild/remote_builder.py

@@ -26,6 +26,7 @@ from flyte._image import (
     UVScript,
 )
 from flyte._internal.imagebuild.image_builder import ImageBuilder, ImageChecker
+from flyte._internal.imagebuild.utils import copy_files_to_context
 from flyte._logging import logger
 from flyte.remote import ActionOutputs, Run
 
@@ -169,7 +170,7 @@ def _get_layers_proto(image: Image, context_path: Path) -> "image_definition_pb2
             )
             layers.append(apt_layer)
         elif isinstance(layer, PythonWheels):
-            dst_path = _copy_files_to_context(layer.wheel_dir, context_path)
+            dst_path = copy_files_to_context(layer.wheel_dir, context_path)
             wheel_layer = image_definition_pb2.Layer(
                 python_wheels=image_definition_pb2.PythonWheels(
                     dir=str(dst_path.relative_to(context_path)),
@@ -184,7 +185,7 @@ def _get_layers_proto(image: Image, context_path: Path) -> "image_definition_pb2
             layers.append(wheel_layer)
 
         elif isinstance(layer, Requirements):
-            dst_path = _copy_files_to_context(layer.file, context_path)
+            dst_path = copy_files_to_context(layer.file, context_path)
             requirements_layer = image_definition_pb2.Layer(
                 requirements=image_definition_pb2.Requirements(
                     file=str(dst_path.relative_to(context_path)),
@@ -240,7 +241,7 @@ def _get_layers_proto(image: Image, context_path: Path) -> "image_definition_pb2
         elif isinstance(layer, DockerIgnore):
             shutil.copy(layer.path, context_path)
         elif isinstance(layer, CopyConfig):
-            dst_path = _copy_files_to_context(layer.src, context_path)
+            dst_path = copy_files_to_context(layer.src, context_path)
 
             copy_layer = image_definition_pb2.Layer(
                 copy_config=image_definition_pb2.CopyConfig(
@@ -264,18 +265,5 @@ def _get_layers_proto(image: Image, context_path: Path) -> "image_definition_pb2
     )
 
 
-def _copy_files_to_context(src: Path, context_path: Path) -> Path:
-    if src.is_absolute() or ".." in str(src):
-        dst_path = context_path / str(src.absolute()).replace("/", "./_flyte_abs_context/", 1)
-    else:
-        dst_path = context_path / src
-    dst_path.parent.mkdir(parents=True, exist_ok=True)
-    if src.is_dir():
-        shutil.copytree(src, dst_path, dirs_exist_ok=True)
-    else:
-        shutil.copy(src, dst_path)
-    return dst_path
-
-
 def _get_fully_qualified_image_name(outputs: ActionOutputs) -> str:
     return outputs.pb2.literals[0].value.scalar.primitive.string_value

flyte/_internal/imagebuild/utils.py

@@ -0,0 +1,29 @@
+import shutil
+from pathlib import Path
+
+
+def copy_files_to_context(src: Path, context_path: Path) -> Path:
+    """
+    This helper function ensures that absolute paths that users specify are converted correctly to a path in the
+    context directory. Doing this prevents collisions while ensuring files are available in the context.
+
+    For example, if a user has
+        img.with_requirements(Path("/Users/username/requirements.txt"))
+           .with_requirements(Path("requirements.txt"))
+           .with_requirements(Path("../requirements.txt"))
+
+    copying with this function ensures that the Docker context folder has all three files.
+
+    :param src: The source path to copy
+    :param context_path: The context path where the files should be copied to
+    """
+    if src.is_absolute() or ".." in str(src):
+        dst_path = context_path / str(src.absolute()).replace("/", "./_flyte_abs_context/", 1)
+    else:
+        dst_path = context_path / src
+    dst_path.parent.mkdir(parents=True, exist_ok=True)
+    if src.is_dir():
+        shutil.copytree(src, dst_path, dirs_exist_ok=True)
+    else:
+        shutil.copy(src, dst_path)
+    return dst_path

flyte/_run.py

@@ -512,7 +512,7 @@ class _Runner:
             raise ValueError("Remote task can only be run in remote mode.")
 
         if not isinstance(task, TaskTemplate) and not isinstance(task, LazyEntity):
-            raise TypeError("On Flyte tasks can be run, not generic functions or methods.")
+            raise TypeError(f"On Flyte tasks can be run, not generic functions or methods '{type(task)}'.")
 
         if self._mode == "remote":
             return await self._run_remote(task, *args, **kwargs)

flyte/_version.py

@@ -1,14 +1,7 @@
 # file generated by setuptools-scm
 # don't change, don't track in version control
 
-__all__ = [
-    "__version__",
-    "__version_tuple__",
-    "version",
-    "version_tuple",
-    "__commit_id__",
-    "commit_id",
-]
+__all__ = ["__version__", "__version_tuple__", "version", "version_tuple"]
 
 TYPE_CHECKING = False
 if TYPE_CHECKING:
@@ -16,19 +9,13 @@ if TYPE_CHECKING:
     from typing import Union
 
     VERSION_TUPLE = Tuple[Union[int, str], ...]
-    COMMIT_ID = Union[str, None]
 else:
     VERSION_TUPLE = object
-    COMMIT_ID = object
 
 version: str
 __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
-commit_id: COMMIT_ID
-__commit_id__: COMMIT_ID
 
-__version__ = version = '2.0.0b7'
-__version_tuple__ = version_tuple = (2, 0, 0, 'b7')
-
-__commit_id__ = commit_id = 'g5cfd1e5ec'
+__version__ = version = '2.0.0b9'
+__version_tuple__ = version_tuple = (2, 0, 0, 'b9')

flyte/cli/_get.py

@@ -201,7 +201,7 @@ def logs(
 
     async def _run_log_view(_obj):
         task = asyncio.create_task(
-            _obj.show_logs(
+            _obj.show_logs.aio(
                 max_lines=lines, show_ts=show_ts, raw=not pretty, attempt=attempt, filter_system=filter_system
             )
         )

flyte/cli/_run.py

@@ -116,7 +116,7 @@ class RunTaskCommand(click.Command):
                         "Run",
                         f"[green bold]Created Run: {r.name} [/green bold] "
                         f"(Project: {r.action.action_id.run.project}, Domain: {r.action.action_id.run.domain})\n"
-                        f"➡️  [blue bold]{r.url}[/blue bold]",
+                        f"➡️  [blue bold][link={r.url}]{r.url}[/link][/blue bold]",
                         obj.output_format,
                     )
                 )
@@ -125,7 +125,7 @@ class RunTaskCommand(click.Command):
                         "[dim]Log streaming enabled, will wait for task to start running "
                         "and log stream to be available[/dim]"
                     )
-                    await r.show_logs(max_lines=30, show_ts=True, raw=False)
+                    await r.show_logs.aio(max_lines=30, show_ts=True, raw=False)
 
         asyncio.run(_run())
 
@@ -212,21 +212,27 @@ run = TaskFiles(
 Run a task from a python file.
 
 Example usage:
+
 ```bash
-flyte run --name examples/basics/hello.py my_task --arg1 value1 --arg2 value2
+flyte run --project my-project --domain development hello.py my_task --arg1 value1 --arg2 value2
 ```
-Note: all arguments for the run command are provided right after the `run` command and before the file name.
 
-You can also specify the project and domain using the `--project` and `--domain` options, respectively. These
-options can be set in the config file or passed as command line arguments.
+Arguments to the run command are provided right after the `run` command and before the file name.
+For example, the command above specifies the project and domain.
+
+To run a task locally, use the `--local` flag. This will run the task in the local environment instead of the remote
+Flyte environment:
 
-Note: The arguments for the task are provided after the task name and can be retrieved using `--help`
-Example:
 ```bash
-flyte run --name examples/basics/hello.py my_task --help
+flyte run --local hello.py my_task --arg1 value1 --arg2 value2
 ```
 
-To run a task locally, use the `--local` flag. This will run the task in the local environment instead of the remote
- Flyte environment.
+Other arguments to the run command are listed below.
+
+Arguments for the task itself are provided after the task name and can be retrieved using `--help`. For example:
+
+```bash
+flyte run hello.py my_task --help
+```
 """,
 )

flyte/remote/_action.py

@@ -258,6 +258,7 @@ class Action(ToJSONMixin):
         """
         return self.pb2.id
 
+    @syncify
     async def show_logs(
         self,
         attempt: int | None = None,

flyte/remote/_logs.py

@@ -30,7 +30,7 @@ def _format_line(logline: payload_pb2.LogLine, show_ts: bool, filter_system: boo
         if logline.originator == payload_pb2.LogLineOriginator.SYSTEM:
             return None
     style = style_map.get(logline.originator, "")
-    if "flyte" in logline.message and "flyte.errors" not in logline.message:
+    if "[flyte]" in logline.message and "flyte.errors" not in logline.message:
         if filter_system:
             return None
         style = "dim"
@@ -101,7 +101,7 @@ class Logs:
         cls,
         action_id: identifier_pb2.ActionIdentifier,
         attempt: int = 1,
-        retry: int = 3,
+        retry: int = 5,
     ) -> AsyncGenerator[payload_pb2.LogLine, None]:
         """
         Tail the logs for a given action ID and attempt.
@@ -135,7 +135,7 @@ class Logs:
                             f"Log stream not available for action {action_id.name} in run {action_id.run.name}."
                         )
                 else:
-                    await asyncio.sleep(1)
+                    await asyncio.sleep(2)
 
     @classmethod
     async def create_viewer(

flyte/remote/_run.py

@@ -141,6 +141,7 @@ class Run(ToJSONMixin):
         """
         return self.action.watch(cache_data_on_done=cache_data_on_done)
 
+    @syncify
     async def show_logs(
         self,
         attempt: int | None = None,
@@ -149,7 +150,7 @@ class Run(ToJSONMixin):
         raw: bool = False,
         filter_system: bool = False,
     ):
-        await self.action.show_logs(attempt, max_lines, show_ts, raw, filter_system=filter_system)
+        await self.action.show_logs.aio(attempt, max_lines, show_ts, raw, filter_system=filter_system)
 
     @syncify
     async def details(self) -> RunDetails:

flyte/remote/_task.py

@@ -3,9 +3,10 @@ from __future__ import annotations
 import functools
 from dataclasses import dataclass
 from threading import Lock
-from typing import Any, AsyncIterator, Callable, Coroutine, Dict, Iterator, Literal, Optional, Tuple, Union
+from typing import Any, AsyncIterator, Callable, Coroutine, Dict, Iterator, Literal, Optional, Tuple, Union, cast
 
 import rich.repr
+from flyteidl.core import literals_pb2
 from google.protobuf import timestamp
 
 import flyte
@@ -13,6 +14,8 @@ import flyte.errors
 from flyte._cache.cache import CacheBehavior
 from flyte._context import internal_ctx
 from flyte._initialize import ensure_client, get_client, get_common_config
+from flyte._internal.runtime.resources_serde import get_proto_resources
+from flyte._internal.runtime.task_serde import get_proto_retry_strategy, get_proto_timeout, get_security_context
 from flyte._logging import logger
 from flyte._protos.common import identifier_pb2, list_pb2
 from flyte._protos.workflow import task_definition_pb2, task_service_pb2
@@ -64,6 +67,15 @@ class LazyEntity:
                 raise RuntimeError(f"Error downloading the task {self._name}, (check original exception...)")
             return self._task
 
+    @syncify
+    async def override(
+        self,
+        **kwargs: Any,
+    ) -> LazyEntity:
+        task_details = cast(TaskDetails, await self.fetch.aio())
+        task_details.override(**kwargs)
+        return self
+
     async def __call__(self, *args, **kwargs):
         """
         Forwards the call to the underlying task. The entity will be fetched if not already present
@@ -271,19 +283,33 @@ class TaskDetails(ToJSONMixin):
     def override(
         self,
         *,
-        local: Optional[bool] = None,
-        ref: Optional[bool] = None,
         resources: Optional[flyte.Resources] = None,
-        cache: flyte.CacheRequest = "auto",
         retries: Union[int, flyte.RetryStrategy] = 0,
         timeout: Optional[flyte.TimeoutType] = None,
-        reusable: Union[flyte.ReusePolicy, Literal["auto"], None] = None,
         env: Optional[Dict[str, str]] = None,
         secrets: Optional[flyte.SecretRequest] = None,
-        max_inline_io_bytes: int | None = None,
         **kwargs: Any,
     ) -> TaskDetails:
-        raise NotImplementedError
+        if len(kwargs) > 0:
+            raise ValueError(
+                f"ReferenceTasks [{self.name}] do not support overriding with kwargs: {kwargs}, "
+                f"Check the parameters for override method."
+            )
+        template = self.pb2.spec.task_template
+        if secrets:
+            template.security_context.CopyFrom(get_security_context(secrets))
+        if template.HasField("container"):
+            if env:
+                template.container.env.clear()
+                template.container.env.extend([literals_pb2.KeyValuePair(key=k, value=v) for k, v in env.items()])
+            if resources:
+                template.container.resources.CopyFrom(get_proto_resources(resources))
+        if retries:
+            template.metadata.retries.CopyFrom(get_proto_retry_strategy(retries))
+        if timeout:
+            template.metadata.timeout.CopyFrom(get_proto_timeout(timeout))
+
+        return self
 
     def __rich_repr__(self) -> rich.repr.Result:
         """

{flyte-2.0.0b7.dist-info → flyte-2.0.0b9.dist-info}/METADATA

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

{flyte-2.0.0b7.dist-info → flyte-2.0.0b9.dist-info}/RECORD

@@ -8,7 +8,7 @@ flyte/_environment.py,sha256=6ks0lkvGt4oSqM5EFPFlhWC3eoUghxUvCn0wstcAD2E,3713
 flyte/_excepthook.py,sha256=nXts84rzEg6-7RtFarbKzOsRZTQR4plnbWVIFMAEprs,1310
 flyte/_group.py,sha256=7o1j16sZyUmYB50mOiq1ui4TBAKhRpDqLakV8Ya1kw4,803
 flyte/_hash.py,sha256=Of_Zl_DzzzF2jp4ZsLm-3o-xJFCCJ8_GubmLI1htx78,504
-flyte/_image.py,sha256=dSChbZeXBSz77wN-AjcXyWLkVUR0kqOtjUuNsflayJM,37310
+flyte/_image.py,sha256=m7wuVTS5xGQrYgS58hEBXJ8GY7YcYMvgPtNp3fRm2fA,37392
 flyte/_initialize.py,sha256=xKl_LYMluRt21wWqa6RTKuLo0_DCbSaTfUk27_brtNk,18232
 flyte/_interface.py,sha256=1B9zIwFDjiVp_3l_mk8EpA4g3Re-6DUBEBi9z9vDvPs,3504
 flyte/_logging.py,sha256=QrT4Z30C2tsZ-yIojisQODTuq6Y6zSJYuTrLgF58UYc,3664
@@ -17,7 +17,7 @@ flyte/_pod.py,sha256=--72b0c6IkOEbBwZPLmgl-ll-j7ECfG-kh75LzBnNN8,1068
 flyte/_resources.py,sha256=L2JuvQDlMo1JLJeUmJPRwtWbunhR2xJEhFgQW5yc72c,9690
 flyte/_retry.py,sha256=rfLv0MvWxzPByKESTglEmjPsytEAKiIvvmzlJxXwsfE,941
 flyte/_reusable_environment.py,sha256=f8Y1GilUwGcXH4n2Fckrnx0SrZmhk3nCfoe-TqUKivI,3740
-flyte/_run.py,sha256=7wT3JyCqZKzM5ZDtM_Xk_KpeafpxGdLA85Adx1Nv3Q8,25662
+flyte/_run.py,sha256=LEZQFhIUf9retXvNrzPN2HV-LkTX77dK7YmL8L0zQmk,25678
 flyte/_secret.py,sha256=89VIihdXI03irHb217GMfipt7jzXBafm17YYmyv6gHo,3245
 flyte/_task.py,sha256=FUqGDtDmhOVPdv-UVko4h0oecoAcc3JZKu8S__cwUpY,19805
 flyte/_task_environment.py,sha256=hblOR-B_Cc6ibpFkedZSyN6V9vi0_vZ0132YKN5D4EM,9826
@@ -25,7 +25,7 @@ flyte/_task_plugins.py,sha256=9MH3nFPOH_e8_92BT4sFk4oyAnj6GJFvaPYWaraX7yE,1037
 flyte/_timeout.py,sha256=zx5sFcbYmjJAJbZWSGzzX-BpC9HC7Jfs35T7vVhKwkk,1571
 flyte/_tools.py,sha256=tWb0sx3t3mm4jbaQVjCTc9y39oR_Ibo3z_KHToP3Lto,966
 flyte/_trace.py,sha256=SSE1nzUgmVTS2xFNtchEOjEjlRavMOIInasXzY8i9lU,4911
-flyte/_version.py,sha256=1YvgXBLYYXUpymS5IKos1Ax9Mkky98AhIJ8qBmjaQ0w,720
+flyte/_version.py,sha256=vlsu0IebyPlbSFljl41fdZ0fVU1AjfW2r3yprWbGAY0,519
 flyte/errors.py,sha256=z28rhbNmJF5Ie7quQWtoSL4K5p_tC3QjZDIZTupNQFw,6395
 flyte/extend.py,sha256=GB4ZedGzKa30vYWRVPOdxEeK62xnUVFY4z2tD6H9eEw,376
 flyte/models.py,sha256=2TgfrkPPgcnnk1P_MO5SEmOYAUbsMKl3gxIDwhW2yEU,15674
@@ -53,9 +53,10 @@ flyte/_internal/controllers/remote/_core.py,sha256=R-gm0tFzsdvyCbrPs0zikXCnzyaTe
 flyte/_internal/controllers/remote/_informer.py,sha256=w4p29_dzS_ns762eNBljvnbJLgCm36d1Ogo2ZkgV1yg,14418
 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=tTA5Wg479ThCB2qwttrZcAqUWBjzzOVUwVg1wNKY4u4,20508
+flyte/_internal/imagebuild/docker_builder.py,sha256=KaaCvVFOY4XNMYMr3LAkDCV-mIzEQPr06UxcwhVD234,21321
 flyte/_internal/imagebuild/image_builder.py,sha256=dXBXl62qcPabus6dR3eP8P9mBGNhpZHZ2Xm12AymKkk,11150
-flyte/_internal/imagebuild/remote_builder.py,sha256=oP8JgnkRgkEMvIdcakxvXjOZiKEYyt5h6CvgEbakSAM,11016
+flyte/_internal/imagebuild/remote_builder.py,sha256=S83QZc5nblQaEyTKn2vxQdvLBIDtvNhDDKRjjjFJ-to,10622
+flyte/_internal/imagebuild/utils.py,sha256=_ULn4jzoffXbuFpB-o_Lro-PvZ9KObYgtELC31NXsgM,1160
 flyte/_internal/resolvers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 flyte/_internal/resolvers/_task_module.py,sha256=jwy1QYygUK7xmpCZLt1SPTfJCkfox3Ck3mTlTsm66UI,1973
 flyte/_internal/resolvers/common.py,sha256=ADQLRoyGsJ4vuUkitffMGrMKKjy0vpk6X53g4FuKDLc,993
@@ -163,10 +164,10 @@ flyte/cli/_create.py,sha256=tXDhZAdsxvpQ4ngO4U-PhyYwzQ3Hi2AJrNtZ3eqJptQ,5541
 flyte/cli/_delete.py,sha256=VTmXv09PBjkdtyl23mbSjIQQlN7Y1AI_bO0GkHP-f9E,546
 flyte/cli/_deploy.py,sha256=sDbO8gobXR4O0Vlp3RKEH-kBSr25BiXJIoHVwau9Occ,8877
 flyte/cli/_gen.py,sha256=ni3E65_wSBc9x5NNbq1REuxfZCJz-ioLMVQnZIgwyYg,5745
-flyte/cli/_get.py,sha256=cnvK5gh_HpISyfXMfn-srqlhyt2xnxZL3ZsGu7ETx9E,10339
+flyte/cli/_get.py,sha256=lzjg62at9oAswuOBORlg5B-LSRrNJgC6C2UViQuzBgk,10343
 flyte/cli/_option.py,sha256=oC1Gs0u0UrOC1SsrFo-iCuAkqQvI1wJWCdjYXA9rW4Q,1445
 flyte/cli/_params.py,sha256=8Gj8UYGHwu-SUXGWCTRX5QsVf19NiajhaUMMae6FF9o,19466
-flyte/cli/_run.py,sha256=z8sZUa03Tx1TFBCwGBrdwq2sQaDr1VTAlMH5miZaEdY,7764
+flyte/cli/_run.py,sha256=qf_E60ZMT06g6j50P1_q5_y0uslIF9FQZE3s4PHycmg,7787
 flyte/cli/main.py,sha256=t5Ivjipd6bVHIGjRBGwkeP577j59ASq9c1wgoNf3h2c,5334
 flyte/config/__init__.py,sha256=MiwEYK5Iv7MRR22z61nzbsbvZ9Q6MdmAU_g9If1Pmb8,144
 flyte/config/_config.py,sha256=WElU--Kw4MM9zx1v-rLD8qYu2T5Zk0-1QbTpkEc27bc,10779
@@ -182,15 +183,15 @@ flyte/io/_dataframe/__init__.py,sha256=SDgNw45uf7m3cHhbUCOA3V3-5A2zSKgPcsWriRLwd
 flyte/io/_dataframe/basic_dfs.py,sha256=weQ8EfzdU-LcKi8Eebq1AiATVS1fGdfcbqtCDOrVLos,7728
 flyte/io/_dataframe/dataframe.py,sha256=uecLLjaAuLyYta2d4Tkk-DjxuHkzZjFUBbvMapPM7R8,51554
 flyte/remote/__init__.py,sha256=y9eke9JzEJkygk8eKZjSj44fJGlyepuW4i-j6lbCAPY,617
-flyte/remote/_action.py,sha256=9Skj2Loroh2ZGarxNHUdljxKgwqR03Qx_ohxaMa4hYA,23936
+flyte/remote/_action.py,sha256=aLaX0-mfFNjOPAB1pskxLbgrljpbvXuvdpGCpU2k7Go,23949
 flyte/remote/_common.py,sha256=2XLLxWL1NjwfdPQUhOfYn3zMrg-yGNfi6NYIaqHutUA,862
 flyte/remote/_console.py,sha256=avmELJPx8nQMAVPrHlh6jEIRPjrMwFpdZjJsWOOa9rE,660
 flyte/remote/_data.py,sha256=zYXXlqEvPdsC44Gm7rP_hQjRgVe3EFfhZNEWKF0p4MQ,6163
-flyte/remote/_logs.py,sha256=aDG18-uPVb2J3PxmqmAY1C0Z4Iv1P1agg-iF4nSQR4U,6709
+flyte/remote/_logs.py,sha256=t4H7DEZDYJC9Vx7oJ7R7m4Z56bWBAjm9ylU4UP1hKq0,6711
 flyte/remote/_project.py,sha256=IbkxKRAvZunKLIwpmcreA4O-0GxWC0KPC62WSYvsK34,2868
-flyte/remote/_run.py,sha256=1QGmXJvUhmuw08zMkT2Eo9ISoaB8AF12eH5G7BBEDGs,10001
+flyte/remote/_run.py,sha256=FtIIMNODQGyc6oYLmuI_ku3bUxhDFIE-UU7ecMNZvBg,10018
 flyte/remote/_secret.py,sha256=CF9WiZKeMJaUNeIawVPf8XHk9OjFt2wc0m7S9ZOdGbE,4399
-flyte/remote/_task.py,sha256=JQbZ8FriV-y_zBer9_H6gJM2AcngY7C_aV-67eJI9WU,15585
+flyte/remote/_task.py,sha256=23sDeYYtcvqNqK51pRB2hmTeBxAmKr1aIbrR6rpKqKg,16693
 flyte/remote/_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 flyte/remote/_client/_protocols.py,sha256=JyBWHs5WsVOxEDUyG9X7wPLDzzzjkoaNhJlU-X4YlN0,5599
 flyte/remote/_client/controlplane.py,sha256=FsOfj4rO4MIMnYrpAT53F8q588VVf5t4sDuwoPuc840,3102
@@ -229,10 +230,10 @@ 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=Tas_OXYddOi0nDuORjqan2SkJ96wKD8937I2l1bo8vk,97916
 flyte/types/_utils.py,sha256=pbts9E1_2LTdLygAY0UYTLYJ8AsN3BZyviSXvrtcutc,2626
-flyte-2.0.0b7.data/scripts/runtime.py,sha256=8U_tIhdF8phBLDFr0U1xv92OKK5RquiaTJFv8v_fIgQ,5537
-flyte-2.0.0b7.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-flyte-2.0.0b7.dist-info/METADATA,sha256=Rkv8GGIoqC6GIkYpy8mpFHoug7XBU5wldwr4YnUfDk4,10004
-flyte-2.0.0b7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-flyte-2.0.0b7.dist-info/entry_points.txt,sha256=MIq2z5dBurdCJfpXfMKzgBv7sJOakKRYxr8G0cMiTrg,75
-flyte-2.0.0b7.dist-info/top_level.txt,sha256=7dkyFbikvA12LEZEqawx8oDG1CMod6hTliPj7iWzgYo,6
-flyte-2.0.0b7.dist-info/RECORD,,
+flyte-2.0.0b9.data/scripts/runtime.py,sha256=8U_tIhdF8phBLDFr0U1xv92OKK5RquiaTJFv8v_fIgQ,5537
+flyte-2.0.0b9.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+flyte-2.0.0b9.dist-info/METADATA,sha256=bXk7WHmum_slHtgZFNlOoGK__r47J3LW6HVsDz_t_YE,10004
+flyte-2.0.0b9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+flyte-2.0.0b9.dist-info/entry_points.txt,sha256=MIq2z5dBurdCJfpXfMKzgBv7sJOakKRYxr8G0cMiTrg,75
+flyte-2.0.0b9.dist-info/top_level.txt,sha256=7dkyFbikvA12LEZEqawx8oDG1CMod6hTliPj7iWzgYo,6
+flyte-2.0.0b9.dist-info/RECORD,,