flyte 0.2.0b14__py3-none-any.whl → 0.2.0b15__py3-none-any.whl

flyte/models.py

@@ -5,7 +5,7 @@ import os
 import pathlib
 import tempfile
 from dataclasses import dataclass, field, replace
-from typing import TYPE_CHECKING, Any, Callable, Dict, Literal, Optional, Tuple, Type
+from typing import TYPE_CHECKING, Any, Callable, ClassVar, Dict, Literal, Optional, Tuple, Type
 
 import rich.repr
 
@@ -15,6 +15,8 @@ from flyte._logging import logger
 from flyte._utils.helpers import base36_encode
 
 if TYPE_CHECKING:
+    from flyteidl.core import literals_pb2
+
     from flyte._internal.imagebuild.image_builder import ImageCache
     from flyte.report import Report
 
@@ -231,6 +233,13 @@ class Checkpoints:
     checkpoint_path: str | None
 
 
+class _has_default:
+    """
+    A marker class to indicate that a specific input has a default value or not.
+    This is used to determine if the input is required or not.
+    """
+
+
 @dataclass(frozen=True)
 class NativeInterface:
     """
@@ -240,7 +249,14 @@ class NativeInterface:
 
     inputs: Dict[str, Tuple[Type, Any]]
     outputs: Dict[str, Type]
-    docstring: Optional[Docstring] = field(default=None)
+    docstring: Optional[Docstring] = None
+
+    # This field is used to indicate that the task has a default value for the input, but already in the
+    # remote form.
+    _remote_defaults: Optional[Dict[str, literals_pb2.Literal]] = field(default=None, repr=False)
+
+    has_default: ClassVar[Type[_has_default]] = _has_default  # This can be used to indicate if a specific input
+    # has a default value or not, in the case when the default value is not known. An example would be remote tasks.
 
     def has_outputs(self) -> bool:
         """
@@ -248,12 +264,31 @@ class NativeInterface:
         """
         return self.outputs is not None and len(self.outputs) > 0
 
+    def num_required_inputs(self) -> int:
+        """
+        Get the number of required inputs for the task. This is used to determine how many inputs are required for the
+        task execution.
+        """
+        return sum(1 for t in self.inputs.values() if t[1] is inspect.Parameter.empty)
+
     @classmethod
-    def from_types(cls, inputs: Dict[str, Type], outputs: Dict[str, Type]) -> NativeInterface:
+    def from_types(
+        cls,
+        inputs: Dict[str, Tuple[Type, Type[_has_default] | Type[inspect._empty]]],
+        outputs: Dict[str, Type],
+        default_inputs: Optional[Dict[str, literals_pb2.Literal]] = None,
+    ) -> NativeInterface:
         """
         Create a new NativeInterface from the given types. This is used to create a native interface for the task.
+        :param inputs: A dictionary of input names and their types and a value indicating if they have a default value.
+        :param outputs: A dictionary of output names and their types.
+        :param default_inputs: Optional dictionary of default inputs for remote tasks.
+        :return: A NativeInterface object with the given inputs and outputs.
         """
-        return cls(inputs={k: (v, inspect.Parameter.empty) for k, v in inputs.items()}, outputs=outputs)
+        for k, v in inputs.items():
+            if v[1] is cls.has_default and (default_inputs is None or k not in default_inputs):
+                raise ValueError(f"Input {k} has a default value but no default input provided for remote task.")
+        return cls(inputs=inputs, outputs=outputs, _remote_defaults=default_inputs)
 
     @classmethod
     def from_callable(cls, func: Callable) -> NativeInterface:
@@ -301,7 +336,10 @@ class NativeInterface:
                 tp = tpe[0] if isinstance(tpe[0], str) else tpe[0].__name__
                 i += f"{key}: {tp}"
                 if tpe[1] is not inspect.Parameter.empty:
-                    i += f" = {tpe[1]}"
+                    if tpe[1] is self.has_default:
+                        i += " = ..."
+                    else:
+                        i += f" = {tpe[1]}"
         i += ")"
         if self.outputs:
             initial = True

flyte/remote/_data.py

@@ -16,6 +16,7 @@ from flyteidl.service import dataproxy_pb2
 from google.protobuf import duration_pb2
 
 from flyte._initialize import CommonInit, ensure_client, get_client, get_common_config
+from flyte._logging import make_hyperlink
 from flyte.errors import InitializationError, RuntimeSystemError
 
 _UPLOAD_EXPIRES_IN = timedelta(seconds=60)
@@ -89,7 +90,7 @@ async def _upload_single_file(
             raise RuntimeSystemError(e.code().value, f"Failed to get signed url for {fp}.")
     except Exception as e:
         raise RuntimeSystemError(type(e).__name__, f"Failed to get signed url for {fp}.") from e
-    logger.debug(f"Uploading to signed url {resp.signed_url} for {fp}")
+    logger.debug(f"Uploading to {make_hyperlink('signed url', resp.signed_url)} for {fp}")
     extra_headers = get_extra_headers_for_protocol(resp.native_url)
     extra_headers.update(resp.headers)
     encoded_md5 = b64encode(md5_bytes)

flyte/remote/_run.py

@@ -43,7 +43,10 @@ def _action_time_phase(action: run_definition_pb2.Action | run_definition_pb2.Ac
         yield "run_time", f"{(datetime.now(timezone.utc) - start_time).seconds} secs"
     yield "phase", run_definition_pb2.Phase.Name(action.status.phase)
     if isinstance(action, run_definition_pb2.ActionDetails):
-        yield "error", action.error_info if action.HasField("error_info") else "NA"
+        yield (
+            "error",
+            f"{action.error_info.kind}: {action.error_info.message}" if action.HasField("error_info") else "NA",
+        )
 
 
 def _action_rich_repr(action: run_definition_pb2.Action) -> rich.repr.Result:
@@ -78,7 +81,7 @@ def _action_details_rich_repr(action: run_definition_pb2.ActionDetails) -> rich.
     yield "task_type", action.resolved_task_spec.task_template.type
     yield "task_version", action.resolved_task_spec.task_template.id.version
     yield "attempts", action.attempts
-    yield "error_info", action.error_info if action.HasField("error_info") else "NA"
+    yield "error", f"{action.error_info.kind}: {action.error_info.message}" if action.HasField("error_info") else "NA"
     yield "phase", run_definition_pb2.Phase.Name(action.status.phase)
     yield "group", action.metadata.group
     yield "parent", action.metadata.parent
@@ -639,7 +642,7 @@ class Action:
                             console.print(f"[bold green]Run '{self.name}' completed successfully.[/bold green]")
                         else:
                             console.print(
-                                f"[bold red]Run '{self.name}' exited unsuccessfully in state {ad.phase}"
+                                f"[bold red]Run '{self.name}' exited unsuccessfully in state {ad.phase} "
                                 f"with error: {ad.error_info}[/bold red]"
                             )
                         break
@@ -667,6 +670,8 @@ class Action:
         Rich representation of the Action object.
         """
         yield from _action_rich_repr(self.pb2)
+        if self._details:
+            yield from self._details.__rich_repr__()
 
     def __repr__(self) -> str:
         """

flyte/remote/_task.py

@@ -162,6 +162,20 @@ class TaskDetails:
         """
         return self.pb2.spec.task_template.type
 
+    @property
+    def default_input_args(self) -> Tuple[str, ...]:
+        """
+        The default input arguments of the task.
+        """
+        return tuple(x.name for x in self.pb2.spec.default_inputs)
+
+    @property
+    def required_args(self) -> Tuple[str, ...]:
+        """
+        The required input arguments of the task.
+        """
+        return tuple(x for x, _ in self.interface.inputs.items() if x not in self.default_input_args)
+
     @functools.cached_property
     def interface(self) -> NativeInterface:
         """
@@ -169,7 +183,7 @@ class TaskDetails:
         """
         import flyte.types as types
 
-        return types.guess_interface(self.pb2.spec.task_template.interface)
+        return types.guess_interface(self.pb2.spec.task_template.interface, default_inputs=self.pb2.spec.default_inputs)
 
     @property
     def cache(self) -> flyte.Cache:
@@ -206,6 +220,19 @@ class TaskDetails:
         """
         Forwards the call to the underlying task. The entity will be fetched if not already present
         """
+        # TODO support kwargs, for this we need ordered inputs to be stored in the task spec.
+        if len(args) > 0:
+            raise flyte.errors.ReferenceTaskError(
+                f"Reference task {self.name} does not support positional arguments"
+                f"currently. Please use keyword arguments."
+            )
+        if len(self.required_args) > 0:
+            if len(args) + len(kwargs) < len(self.required_args):
+                raise ValueError(
+                    f"Task {self.name} requires at least {self.required_args} arguments, "
+                    f"but only received args:{args}  kwargs{kwargs}."
+                )
+
         ctx = internal_ctx()
         if ctx.is_task_context():
             # If we are in a task context, that implies we are executing a Run.
@@ -238,6 +265,11 @@ class TaskDetails:
         """
         Rich representation of the task.
         """
+        yield "friendly_name", self.pb2.spec.short_name
+        yield "environment", self.pb2.spec.environment
+        yield "default_inputs_keys", self.default_input_args
+        yield "required_args", self.required_args
+        yield "raw_default_inputs", [str(x) for x in self.pb2.spec.default_inputs]
         yield "project", self.pb2.task_id.project
         yield "domain", self.pb2.task_id.domain
         yield "name", self.name

flyte/storage/__init__.py

@@ -4,6 +4,7 @@ __all__ = [
     "S3",
     "Storage",
     "get",
+    "get_configured_fsspec_kwargs",
     "get_random_local_directory",
     "get_random_local_path",
     "get_stream",
@@ -18,6 +19,7 @@ __all__ = [
 from ._config import ABFS, GCS, S3, Storage
 from ._storage import (
     get,
+    get_configured_fsspec_kwargs,
     get_random_local_directory,
     get_random_local_path,
     get_stream,

flyte/storage/_storage.py

@@ -62,40 +62,51 @@ def get_random_local_directory() -> pathlib.Path:
     return _dir
 
 
-def get_underlying_filesystem(
-    protocol: typing.Optional[str] = None,
-    anonymous: bool = False,
-    path: typing.Optional[str] = None,
-    **kwargs,
-) -> fsspec.AbstractFileSystem:
-    if protocol is None:
-        # If protocol is None, get it from the path
-        protocol = get_protocol(path)
-
-    storage_config = get_storage()
-    if storage_config:
-        kwargs = storage_config.get_fsspec_kwargs(anonymous, **kwargs)
-    elif protocol:
+def get_configured_fsspec_kwargs(
+    protocol: typing.Optional[str] = None, anonymous: bool = False
+) -> typing.Dict[str, typing.Any]:
+    if protocol:
         match protocol:
             case "s3":
                 # If the protocol is s3, we can use the s3 filesystem
                 from flyte.storage import S3
 
-                kwargs = S3.auto().get_fsspec_kwargs(anonymous=anonymous, **kwargs)
+                return S3.auto().get_fsspec_kwargs(anonymous=anonymous)
             case "gs":
                 # If the protocol is gs, we can use the gs filesystem
                 from flyte.storage import GCS
 
-                kwargs = GCS.auto().get_fsspec_kwargs(anonymous=anonymous, **kwargs)
+                return GCS.auto().get_fsspec_kwargs(anonymous=anonymous)
             case "abfs" | "abfss":
                 # If the protocol is abfs or abfss, we can use the abfs filesystem
                 from flyte.storage import ABFS
 
-                kwargs = ABFS.auto().get_fsspec_kwargs(anonymous=anonymous, **kwargs)
+                return ABFS.auto().get_fsspec_kwargs(anonymous=anonymous)
             case _:
-                pass
+                return {}
+
+    # If no protocol, return args from storage config if set
+    storage_config = get_storage()
+    if storage_config:
+        return storage_config.get_fsspec_kwargs(anonymous)
+
+    return {}
+
+
+def get_underlying_filesystem(
+    protocol: typing.Optional[str] = None,
+    anonymous: bool = False,
+    path: typing.Optional[str] = None,
+    **kwargs,
+) -> fsspec.AbstractFileSystem:
+    if protocol is None:
+        # If protocol is None, get it from the path
+        protocol = get_protocol(path)
+
+    configured_kwargs = get_configured_fsspec_kwargs(protocol, anonymous=anonymous)
+    configured_kwargs.update(kwargs)
 
-    return fsspec.filesystem(protocol, **kwargs)
+    return fsspec.filesystem(protocol, **configured_kwargs)
 
 
 def _get_anonymous_filesystem(from_path):

flyte/types/_interface.py

@@ -1,11 +1,15 @@
-from typing import Any, Dict, Type, cast
+import inspect
+from typing import Any, Dict, Iterable, Tuple, Type, cast
 
-from flyteidl.core import interface_pb2
+from flyteidl.core import interface_pb2, literals_pb2
 
+from flyte._protos.workflow import common_pb2
 from flyte.models import NativeInterface
 
 
-def guess_interface(interface: interface_pb2.TypedInterface) -> NativeInterface:
+def guess_interface(
+    interface: interface_pb2.TypedInterface, default_inputs: Iterable[common_pb2.NamedParameter] | None = None
+) -> NativeInterface:
     """
     Returns the interface of the task with guessed types, as types may not be present in current env.
     """
@@ -14,12 +18,23 @@ def guess_interface(interface: interface_pb2.TypedInterface) -> NativeInterface:
     if interface is None:
         return NativeInterface({}, {})
 
-    guessed_inputs: Dict[str, Type[Any]] = {}
+    default_input_literals: Dict[str, literals_pb2.Literal] = {}
+    if default_inputs is not None:
+        for param in default_inputs:
+            if param.parameter.HasField("default"):
+                default_input_literals[param.name] = param.parameter.default
+
+    guessed_inputs: Dict[str, Tuple[Type[Any], Any] | Any] = {}
     if interface.inputs is not None and len(interface.inputs.variables) > 0:
-        guessed_inputs = flyte.types.TypeEngine.guess_python_types(cast(dict, interface.inputs.variables))
+        input_types = flyte.types.TypeEngine.guess_python_types(cast(dict, interface.inputs.variables))
+        for name, t in input_types.items():
+            if name not in default_input_literals:
+                guessed_inputs[name] = (t, inspect.Parameter.empty)
+            else:
+                guessed_inputs[name] = (t, NativeInterface.has_default)
 
     guessed_outputs: Dict[str, Type[Any]] = {}
     if interface.outputs is not None and len(interface.outputs.variables) > 0:
         guessed_outputs = flyte.types.TypeEngine.guess_python_types(cast(dict, interface.outputs.variables))
 
-    return NativeInterface.from_types(guessed_inputs, guessed_outputs)
+    return NativeInterface.from_types(guessed_inputs, guessed_outputs, default_input_literals)

{flyte-0.2.0b14.dist-info → flyte-0.2.0b15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flyte
-Version: 0.2.0b14
+Version: 0.2.0b15
 Summary: Add your description here
 Author-email: Ketan Umare <kumare3@users.noreply.github.com>
 Requires-Python: >=3.10
@@ -125,7 +125,7 @@ import flyte
 
 env = flyte.TaskEnvironment(
     name="hello_world",
-    image=flyte.Image.auto().with_apt_packages(...).with_pip_packages(...),
+    image=flyte.Image.from_debian_base().with_apt_packages(...).with_pip_packages(...),
 )
 
 ```

{flyte-0.2.0b14.dist-info → flyte-0.2.0b15.dist-info}/RECORD

@@ -1,42 +1,42 @@
-flyte/__init__.py,sha256=lg45TANKqnZrPZhIOelGabO2rzvJzRIhy1tDX4kqTwo,1555
+flyte/__init__.py,sha256=2OcML9ib7PwGyFRfQqc_i0l4BKHLabLbwBFvB7p1OeU,1699
 flyte/_build.py,sha256=MkgfLAPeL56YeVrGRNZUCZgbwzlEzVP3wLbl5Qru4yk,578
 flyte/_context.py,sha256=K0-TCt-_pHOoE5Xni87_8uIe2vCBOhfNQEtjGT4Hu4k,5239
-flyte/_deploy.py,sha256=Nw1E-jU2AxFMbRG_DWfxiKO6InZCr1Ow2azd6PbKe0E,8987
+flyte/_deploy.py,sha256=4RfrjSniBYbHSKRKQpmACGHPYdlaE4voX4jn-BdcOhA,9280
 flyte/_doc.py,sha256=_OPCf3t_git6UT7kSJISFaWO9cfNzJhhoe6JjVdyCJo,706
 flyte/_docstring.py,sha256=SsG0Ab_YMAwy2ABJlEo3eBKlyC3kwPdnDJ1FIms-ZBQ,1127
-flyte/_environment.py,sha256=BkChtdVhWB3SwMSvetDZ-KiNBgRFlAXgq69PHT4zyG0,2942
+flyte/_environment.py,sha256=dmIFmFLRIklOEYL9gsP2IH3-MYcjHYyyOlqlcf3E6_A,2924
 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=NeBvjCdwFAVGd666ufi1q-YOvhwdTEzAeJl5YBfl0bI,29043
+flyte/_image.py,sha256=_4IxJ5T4IESBRrxT9ZOlnStSUThcYtceGUDLgtCxMg8,29149
 flyte/_initialize.py,sha256=zPcygfpUM24vZ8-6BizPTiDYHeEgXSPbqD4QqJ_pA_A,17831
-flyte/_interface.py,sha256=MP5o_qpIwfBNtAc7zo_cLSjMugsPyanuO6EgUSk4fBE,3644
-flyte/_logging.py,sha256=n3pi1cE_oqA03vQHlqSxexXcjRhCsZg1ODvUSoWlwk4,3382
+flyte/_interface.py,sha256=1B9zIwFDjiVp_3l_mk8EpA4g3Re-6DUBEBi9z9vDvPs,3504
+flyte/_logging.py,sha256=_yNo-Nx2yzh0MLoZGbnIYHGKei4wtQmSGM0lE30Ev7w,3662
 flyte/_map.py,sha256=efPd8O-JKUg1OY3_MzU3KGbhsGYDVRNBwWr0ceNIXhQ,7444
 flyte/_pod.py,sha256=lNaQuWX22QG6Xji7-8GpuKUkqCmVFaRxOVU-eUEa-Vk,637
 flyte/_resources.py,sha256=UOLyEVhdxolvrHhddiBbYdJuE1RkM_l7xeS9G1abe6M,7583
 flyte/_retry.py,sha256=rfLv0MvWxzPByKESTglEmjPsytEAKiIvvmzlJxXwsfE,941
 flyte/_reusable_environment.py,sha256=P4FBATVKAYcIKpdFN98sI8acPyKy8eIGx6V0kUb9YdM,1289
-flyte/_run.py,sha256=kxylo-BTgnZC7FkqvKUtIDeXE6kevLqCS8i0buT70BA,20196
+flyte/_run.py,sha256=32pLVE2UhZOYOHfYH2yKPSzVFkRj1OjjCUxgCYMyeXE,20200
 flyte/_secret.py,sha256=SqIHs6mi8hEkIIBZe3bI9jJsPt65Mt6dV5uh9_op1ME,2392
-flyte/_task.py,sha256=AHxg4lqHfExdSU6adwPiFzAT2EtrLI8mBdRxTUL1RgA,17902
-flyte/_task_environment.py,sha256=J1LFH9Zz1nPhlsrc_rYny1SS3QC1b55X7tRYoTG7Vk4,6815
+flyte/_task.py,sha256=-3VLSROj8g3-ZWzV272Ww7mt5yIeA753kmpNyr75I58,18087
+flyte/_task_environment.py,sha256=1PIuRQ6inH0vMw89iutWKAyPff9qaTEcjk6UR1Gl0cg,6861
 flyte/_timeout.py,sha256=zx5sFcbYmjJAJbZWSGzzX-BpC9HC7Jfs35T7vVhKwkk,1571
 flyte/_tools.py,sha256=JewkQZBR_M85tS6QY8e4xXue75jbOE48nID4ZHnc9jY,632
 flyte/_trace.py,sha256=C788bgoSc3st8kE8Cae2xegnLx2CT6uuRKKfaDrDUys,5122
-flyte/_version.py,sha256=v28x7rXB_3GrL0I5zyhrj-6Ffe8gWhHERwr3KVTgEgw,521
-flyte/errors.py,sha256=lJgSiZb2nZnuTZqBEB9rg-bV_GXVAYNjQFRuKWQskyY,4683
-flyte/models.py,sha256=A85HnTLqInJiMmQKhpl2IXb2Uh6_46tdRrwW2TTzD9o,12908
+flyte/_version.py,sha256=RbabyiAT9BZS4y_-wPKvDqifAniKvrfGunjyicmEHU0,521
+flyte/errors.py,sha256=skXcdexLisFZFcTnUmMvMmuh4Ty96oJkyLKaipzkyeI,4954
+flyte/models.py,sha256=my7Vxo-NK6gHGahyqtHr42wYjPGw0nl2SGBBoSb6hIc,14733
 flyte/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 flyte/_bin/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-flyte/_bin/runtime.py,sha256=09OKQeUwyEDkXF2HtzHeYVJsIMkEzLIO2tVCARvPGJE,5783
+flyte/_bin/runtime.py,sha256=JXh1wgOhCnIMBE7YJY9biepY3VedrGdIFHDfzmz5gjM,5783
 flyte/_cache/__init__.py,sha256=zhdO5UuHQRdzn8GHmSN40nrxfAmI4ihDRuHZM11U84Y,305
 flyte/_cache/cache.py,sha256=ErhWzzJdEjTIuEF4f-r6IBgko-3Al9iUs1Eq4O42TUE,5021
 flyte/_cache/defaults.py,sha256=gzJZW0QJPUfd2OPnGpv3tzIfwPtgFjAKoie3NP1P97U,217
 flyte/_cache/policy_function_body.py,sha256=_AcyN6XKRXq16yV5lWuRJYCIVUlmyPvvWuYRxfU-Ldo,1507
 flyte/_code_bundle/__init__.py,sha256=G7DJTQ0UN_ETvdh55pYcWsTrZJKXEcyQl9iQQNQOBXQ,328
 flyte/_code_bundle/_ignore.py,sha256=Tfaoa62CQVTH17kBHD6Xv6xEh1FhcAyvXivl9m-MEE0,3853
-flyte/_code_bundle/_packaging.py,sha256=_wEozcQTYgqvAAaKQYna9ptvShIMlXk3vEdccwAOYn8,6873
+flyte/_code_bundle/_packaging.py,sha256=5QUuea6kg9s-ebBg7gFAHaxOMchxR5MhTQ8KohWsjPk,6909
 flyte/_code_bundle/_utils.py,sha256=b0s3ZVKSRwaa_2CMTCqt2iRrUvTTW3FmlyqCD9k5BS0,12028
 flyte/_code_bundle/bundle.py,sha256=8T0gcXck6dmg-8L2-0G3B2iNjC-Xwydu806iyKneMMY,8789
 flyte/_internal/__init__.py,sha256=vjXgGzAAjy609YFkAy9_RVPuUlslsHSJBXCLNTVnqOY,136
@@ -46,24 +46,24 @@ flyte/_internal/controllers/_trace.py,sha256=biI-lXSIe3gXuWI-KT6T-jTtojQCQ7BLOHT
 flyte/_internal/controllers/remote/__init__.py,sha256=9_azH1eHLqY6VULpDugXi7Kf1kK1ODqEnsQ_3wM6IqU,1919
 flyte/_internal/controllers/remote/_action.py,sha256=aGmTCciHkx9YSU6gPWWFsgjIan-mgelHfmZM4SzZJS4,5121
 flyte/_internal/controllers/remote/_client.py,sha256=HPbzbfaWZVv5wpOvKNtFXR6COiZDwd1cUJQqi60A7oU,1421
-flyte/_internal/controllers/remote/_controller.py,sha256=qA_uas6Vhk0VEAXNtI2obogtmC-YCr8mTexSnvVA3gg,21184
+flyte/_internal/controllers/remote/_controller.py,sha256=G4gEUl-2f4FqACz3IK7sK0-w93kkJndzkDpbISqo_4w,21247
 flyte/_internal/controllers/remote/_core.py,sha256=UKjiL3dGidXcR1dPzJswn67g38HQmVXV7n0zoX_8AZY,18422
 flyte/_internal/controllers/remote/_informer.py,sha256=StiPcQLLW0h36uEBhKsupMY79EeFCKA3QQzvv2IyvRo,14188
 flyte/_internal/controllers/remote/_service_protocol.py,sha256=B9qbIg6DiGeac-iSccLmX_AL2xUgX4ezNUOiAbSy4V0,1357
 flyte/_internal/imagebuild/__init__.py,sha256=cLXVxkAyFpbdC1y-k3Rb6FRW9f_xpoRQWVn__G9IqKs,354
-flyte/_internal/imagebuild/docker_builder.py,sha256=bkr2fs9jInTq8jqU8ka7NGvp0RPfYhbTfX1RqtqTvvs,13986
-flyte/_internal/imagebuild/image_builder.py,sha256=q-TRcF_t2nS8FqBOtIIdnSo1Jrq378jqzceYDPzne8Q,9902
+flyte/_internal/imagebuild/docker_builder.py,sha256=cvJ_e2hDvLS9SBvdzgNwvSvkQORP6m-_mjMBHZf-m8I,14479
+flyte/_internal/imagebuild/image_builder.py,sha256=mzC0Dz_vUCnOB9ZO4zRMoS0qXFGsw6UBC_xOqU8TVkQ,9785
 flyte/_internal/imagebuild/remote_builder.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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
 flyte/_internal/resolvers/default.py,sha256=nX4DHUYod1nRvEsl_vSgutQVEdExu2xL8pRkyi4VWbY,981
 flyte/_internal/runtime/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-flyte/_internal/runtime/convert.py,sha256=zDMLjmzy84cOcC5rZZA1RbbXOP9irmWHNfCn5HrgprI,11522
+flyte/_internal/runtime/convert.py,sha256=-ZjnymAhAmO6ngNkbhwrxRrvvXD0QQRnZUPptY1pyuo,13777
 flyte/_internal/runtime/entrypoints.py,sha256=Kyi19i7LYk7YM3ZV_Y4FXGt5Pc1tIftGkIDohopblyY,5127
 flyte/_internal/runtime/io.py,sha256=Lgdy4iPjlKjUO-V_AkoPZff6lywaFjZUG-PErRukmx4,4248
 flyte/_internal/runtime/resources_serde.py,sha256=tvMMv3l6cZEt_cfs7zVE_Kqs5qh-_r7fsEPxb6xMxMk,4812
-flyte/_internal/runtime/task_serde.py,sha256=-CiOvrktT2ChdNhdFzgTu9r1Lujk1wkjs8bhmsILnYw,13045
+flyte/_internal/runtime/task_serde.py,sha256=H7cyRCS8eyicnWjQogcFPHxs1uX9FI5bZypUfBSzJOw,13873
 flyte/_internal/runtime/taskrunner.py,sha256=rHWS4t5qgZnzGdGrs0_O0sSs_PVGoE1CNPDb-fTwwmo,7332
 flyte/_internal/runtime/types_serde.py,sha256=EjRh9Yypx9-20XXQprtNgp766LeQVRoYWtY6XPGMZQg,1813
 flyte/_protos/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -135,44 +135,44 @@ flyte/_protos/workflow/task_service_pb2_grpc.py,sha256=whmfmOTiNhz6_CBsXm8aXUCwt
 flyte/_utils/__init__.py,sha256=RByuYa2hSqsKrpLkFfjKspKxH3K7DzfVzTVnZsxkO0A,757
 flyte/_utils/asyn.py,sha256=KeJKarXNIyD16g6oPM0T9cH7JDmh1KY7JLbwo7i0IlQ,3673
 flyte/_utils/async_cache.py,sha256=JtZJmWO62OowJ0QFNl6wryWqh-kuDi76aAASMie87QY,4596
-flyte/_utils/coro_management.py,sha256=_JTt9x9fOc_1OSe03DSheYoKOvlonBB_4WNCS9XSaoU,698
+flyte/_utils/coro_management.py,sha256=wIsul4XY-tQbH9bjqZ3A0jKluE19xSzLlkMeYu_dk_A,934
 flyte/_utils/file_handling.py,sha256=iU4TxW--fCho_Eg5xTMODn96P03SxzF-V-5f-7bZAZY,2233
-flyte/_utils/helpers.py,sha256=45ZC2OSNKS66DkTvif8W8x7MH4KxluvAyn0a92mKRqM,4366
+flyte/_utils/helpers.py,sha256=9N70yzfLF4lLGEEdOv5OcweEpYtrCvZqqhtzkjZUXNY,4779
 flyte/_utils/lazy_module.py,sha256=fvXPjvZLzCfcI8Vzs4pKedUDdY0U_RQ1ZVrp9b8qBQY,1994
 flyte/_utils/org_discovery.py,sha256=C7aJa0LfnWBkDtSU9M7bE60zp27qEhJC58piqOErZ94,2088
 flyte/_utils/uv_script_parser.py,sha256=PxqD8lSMi6xv0uDd1s8LKB2IPZr4ttZJCUweqlyMTKk,1483
 flyte/cli/__init__.py,sha256=M02O-UGqQlA8JJ_jyWRiwQhTNc5CJJ7x9J7fNxTxBT0,52
 flyte/cli/_abort.py,sha256=Ty-63Gtd2PUn6lCuL5AaasfBoPu7TDSU5EQKVbkF4qw,661
-flyte/cli/_common.py,sha256=r7QMoN4uQPQnjYd7TVnpk8RcbWJYCliA4Aa0aMbFDZo,11047
+flyte/cli/_common.py,sha256=QphqiND3LkF3bngngtArZWqdLMe1uiG5K9XRCHwRdhA,10973
 flyte/cli/_create.py,sha256=a75II-xT71SpdopNY14rPuidO5qL0mH1UAwf205sVzQ,4313
 flyte/cli/_delete.py,sha256=VTmXv09PBjkdtyl23mbSjIQQlN7Y1AI_bO0GkHP-f9E,546
-flyte/cli/_deploy.py,sha256=o8xtChouozypfE8M2BtfhrcxMEqjlrxB8JsYxPENNhc,4789
+flyte/cli/_deploy.py,sha256=_7tMgLavjvncdTdrn1sxN4UlnL16vuC1o-k5VefZTxM,4841
 flyte/cli/_gen.py,sha256=vlE5l8UR1zz4RSdaRyUfYFvGR0TLxGcTYcP4dhA3Pvg,5458
 flyte/cli/_get.py,sha256=UBh82YQ3WZF7WPZjW3l3RQG97JZYlgHMzpuiVdtJseg,9896
 flyte/cli/_params.py,sha256=cDeTvjOQP8EydVJUrncLeAxUaHvGorJyDvMSrAxapmM,19466
 flyte/cli/_run.py,sha256=a8Y1XLhti6kaaibcN2AZHjxIauHeXqT9wnXWDKee2ew,7733
-flyte/cli/main.py,sha256=Wv-Bd70f-xANAfRK9III5SYH4-r4ScHqWx8EmZbsoII,4408
+flyte/cli/main.py,sha256=a9VsgyGi-s_FoXL-HsSXYSQfLV_ljCJWnTCriOHWXWI,4382
 flyte/config/__init__.py,sha256=MiwEYK5Iv7MRR22z61nzbsbvZ9Q6MdmAU_g9If1Pmb8,144
 flyte/config/_config.py,sha256=QE3T0W8xOULjJaqDMdMF90f9gFVjGR6h8QPOLsyqjYw,9831
 flyte/config/_internal.py,sha256=Bj0uzn3PYgxKbzM-q2GKXxp7Y6cyzhPzUB-Y2i6cQKo,2836
 flyte/config/_reader.py,sha256=c16jm0_IYxwEAjXENtllLeO_sT5Eg2RNLG4UjnAv_x4,7157
 flyte/connectors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 flyte/extras/__init__.py,sha256=FhB0uK7H1Yo5De9vOuF7UGnezTKncj3u2Wo5uQdWN0g,74
-flyte/extras/_container.py,sha256=7fg43UScXevb9QdiyZLyf0xFrL5L3GkPvUWKQLl4R98,10939
+flyte/extras/_container.py,sha256=bB9wxoU-Xq75LEIoR-MkadbImjCUNRv_Tf-b2pMTct8,10947
 flyte/io/__init__.py,sha256=F7hlpin_1JJjsdFZSn7_jQgltPzsjETX1DCYGz-ELqI,629
 flyte/io/_dir.py,sha256=rih9CY1YjNX05bcAu5LG62Xoyij5GXAlv7jLyVF0je8,15310
 flyte/io/_file.py,sha256=kp5700SKPy5htmMhm4hE2ybb99Ykny1b0Kwm3huCWXs,15572
 flyte/io/_structured_dataset/__init__.py,sha256=69ixVV9OEXiLiQ6SV2S8tEC7dVQe7YTt9NV1OotlG64,4524
-flyte/io/_structured_dataset/basic_dfs.py,sha256=77aYFrFigPC7cjM6UjCbU26REtXmwIBBapFN6KGYOO8,8224
+flyte/io/_structured_dataset/basic_dfs.py,sha256=D0QzcaMBO_R2s9Oi9mDqiykkBp0kgi-eI_A3w92vUkw,7980
 flyte/io/_structured_dataset/structured_dataset.py,sha256=ddRjz36RhNxIy0gakzdLStBzoo4cAOgXbNqiqt5YhMI,52645
 flyte/remote/__init__.py,sha256=h0J9W1yWbvPq2R9HXa_HezJtxHoWl6d9QKQbuuKDMnU,597
 flyte/remote/_console.py,sha256=avmELJPx8nQMAVPrHlh6jEIRPjrMwFpdZjJsWOOa9rE,660
-flyte/remote/_data.py,sha256=9PNyZBWpub1ookQZMEeOs0MDhohOeENrUEIVf6FKDPE,6017
+flyte/remote/_data.py,sha256=EO1ZXGYvhfFDOlU9Q0-tSJqLK4shc7q3nvOPPWJxuTc,6078
 flyte/remote/_logs.py,sha256=EOXg4OS8yYclsT6NASgOLMo0TA2sZpKb2MWZXpWBPuI,6404
 flyte/remote/_project.py,sha256=dTBYqORDAbLvh9WnPO1Ytuzw2vxNYZwwNsKE2_b0o14,2807
-flyte/remote/_run.py,sha256=9euHjYRX-xyxXuhn0MunYb9dmgl0FMU3a-FZNjJA4F8,31057
+flyte/remote/_run.py,sha256=MQRxd6Ib_Si2aGV83H4P1ZR3QiDpec_gwsAtjjkp8qY,31247
 flyte/remote/_secret.py,sha256=l5xeMS83uMcWWeSSTRsSZUNhS0N--1Dze09C-thSOQs,4341
-flyte/remote/_task.py,sha256=7C8iKyeeiCMuePjDIihRhkWOtqRDmRJDq1e5lRK2s6U,12781
+flyte/remote/_task.py,sha256=7e5Y0K8vd8TC2J6Nb0nOCUPaUh8IiV3kKfgsCP9CcGw,14227
 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
@@ -197,22 +197,22 @@ flyte/remote/_client/auth/_grpc_utils/default_metadata_interceptor.py,sha256=IoM
 flyte/report/__init__.py,sha256=yLbeUxYaVaDlgBod3Oh34zGBSotl1UlXq1vUkb9q7cs,152
 flyte/report/_report.py,sha256=jZXl-dqkZHiVAPBuUCmE4e1vE9FpHLkermOeZehi8Tc,5175
 flyte/report/_template.html,sha256=YehmLJG3QMYQ10UT1YZBu2ncVmAJ4iyqVp5hF3sXRAs,3458
-flyte/storage/__init__.py,sha256=kkOyqBXJVZSKt3ALitGn9zK6bkchy1-c3TOPB4QhPxk,499
+flyte/storage/__init__.py,sha256=0tcI9qtIVf0Fxczkno03vpwBDVlKMDSNN38uxMTH1bE,569
 flyte/storage/_config.py,sha256=xVibWJaioOnkeTb_M30azgiUe1jvmQaOWRZEkpdoTao,8680
 flyte/storage/_remote_fs.py,sha256=kM_iszbccjVD5VtVdgfkl1FHS8NPnY__JOo_CPQUE4c,1124
-flyte/storage/_storage.py,sha256=mBy7MKII2M1UTVm_EUUDwVb7uT1_AOPzQr2wCJ-fgW0,9873
+flyte/storage/_storage.py,sha256=AQ_ubpYwJjRdEYV-qU4ttXDZuuT28t6-S5eFQD5D9sQ,10175
 flyte/storage/_utils.py,sha256=8oLCM-7D7JyJhzUi1_Q1NFx8GBUPRfou0T_5tPBmPbE,309
 flyte/syncify/__init__.py,sha256=WgTk-v-SntULnI55CsVy71cxGJ9Q6pxpTrhbPFuouJ0,1974
 flyte/syncify/_api.py,sha256=fh7Dlh0cZGELnWDh5lFmxVu-_cqlfsu9XLwkiuDRseE,14672
 flyte/types/__init__.py,sha256=9310PRtVrwJePwEPeoUO0HPyIkgaja7-Dar_QlE_MUI,1745
-flyte/types/_interface.py,sha256=mY7mb8v2hJPGk7AU99gdOWl4_jArA1VFtjYGlE31SK0,953
+flyte/types/_interface.py,sha256=5y9EC5r897xz03Hh0ltF8QVGKMfMfAznws-hKSEO4Go,1677
 flyte/types/_pickle.py,sha256=PjdR66OTDMZ3OYq6GvM_Ua0cIo5t2XQaIjmpJ9xo4Ys,4050
 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=Kk5g1nZubh2A4KWbvGBf5A-aZTvbTtjrY1Bg1_GOPV4,96643
 flyte/types/_utils.py,sha256=pbts9E1_2LTdLygAY0UYTLYJ8AsN3BZyviSXvrtcutc,2626
-flyte-0.2.0b14.dist-info/METADATA,sha256=1I-_oekIUsZRu7qdLO-ee81A9jMn-ZOJfmRG9YUhSUc,5838
-flyte-0.2.0b14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-flyte-0.2.0b14.dist-info/entry_points.txt,sha256=MIq2z5dBurdCJfpXfMKzgBv7sJOakKRYxr8G0cMiTrg,75
-flyte-0.2.0b14.dist-info/top_level.txt,sha256=7dkyFbikvA12LEZEqawx8oDG1CMod6hTliPj7iWzgYo,6
-flyte-0.2.0b14.dist-info/RECORD,,
+flyte-0.2.0b15.dist-info/METADATA,sha256=8EpO_eWz-jwLENNAJAOph4ku1Z3h6-K7OG5W9tv1y3M,5850
+flyte-0.2.0b15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+flyte-0.2.0b15.dist-info/entry_points.txt,sha256=MIq2z5dBurdCJfpXfMKzgBv7sJOakKRYxr8G0cMiTrg,75
+flyte-0.2.0b15.dist-info/top_level.txt,sha256=7dkyFbikvA12LEZEqawx8oDG1CMod6hTliPj7iWzgYo,6
+flyte-0.2.0b15.dist-info/RECORD,,