flyte 0.2.0b10__py3-none-any.whl → 0.2.0b11__py3-none-any.whl

flyte/__init__.py

@@ -25,6 +25,7 @@ __all__ = [
     "Device",
     "Environment",
     "Image",
+    "PodTemplate",
     "Resources",
     "RetryStrategy",
     "ReusePolicy",
@@ -53,6 +54,7 @@ from ._group import group
 from ._image import Image
 from ._initialize import init, init_from_config
 from ._map import map
+from ._pod import PodTemplate
 from ._resources import GPU, TPU, Device, Resources
 from ._retry import RetryStrategy
 from ._reusable_environment import ReusePolicy

flyte/_bin/runtime.py

@@ -76,13 +76,17 @@ def main(
 ):
     sys.path.insert(0, ".")
 
+    import flyte
     import flyte._utils as utils
     from flyte._initialize import initialize_in_cluster
     from flyte._internal.controllers import create_controller
     from flyte._internal.imagebuild.image_builder import ImageCache
     from flyte._internal.runtime.entrypoints import load_and_run_task
+    from flyte._logging import logger
     from flyte.models import ActionID, Checkpoints, CodeBundle, RawDataPath
 
+    logger.info(f"Initializing flyte runtime - version {flyte.__version__}")
+
     assert org, "Org is required for now"
     assert project, "Project is required"
     assert domain, "Domain is required"
@@ -98,15 +102,14 @@ def main(
     # This detection of api key is a hack for now.
     controller_kwargs: dict[str, Any] = {"insecure": False}
     if api_key := os.getenv(_UNION_EAGER_API_KEY_ENV_VAR):
-        from flyte._logging import logger
-
-        logger.warning(f"Using api key {api_key}")
+        logger.info("Using api key from environment")
         controller_kwargs["api_key"] = api_key
     else:
         ep = os.environ.get(ENDPOINT_OVERRIDE, "host.docker.internal:8090")
         controller_kwargs["endpoint"] = ep
         if "localhost" in ep or "docker" in ep:
             controller_kwargs["insecure"] = True
+        logger.debug(f"Using controller endpoint: {ep} with kwargs: {controller_kwargs}")
 
     bundle = CodeBundle(tgz=tgz, pkl=pkl, destination=dest, computed_version=version)
     initialize_in_cluster()

flyte/_initialize.py

@@ -168,6 +168,8 @@ async def init(
     """
     from flyte._utils import get_cwd_editable_install
 
+    from ._utils.org_discovery import org_from_endpoint
+
     interactive_mode = ipython_check()
 
     initialize_logger(enable_rich=interactive_mode)
@@ -206,12 +208,14 @@ async def init(
             domain=domain,
             client=client,
             storage=storage,
-            org=org,
+            org=org or org_from_endpoint(endpoint),
         )
 
 
 @syncify
-async def init_from_config(path_or_config: str | Config | None = None, root_dir: Path | None = None) -> None:
+async def init_from_config(
+    path_or_config: str | Config | None = None, root_dir: Path | None = None, log_level: int | None = None
+) -> None:
     """
     Initialize the Flyte system using a configuration file or Config object. This method should be called before any
     other Flyte remote API methods are called. Thread-safe implementation.
@@ -221,6 +225,8 @@ async def init_from_config(path_or_config: str | Config | None = None, root_dir:
         files like config etc. For example if one uses the copy-style=="all", it is essential to determine the
         root directory for the current project. If not provided, it defaults to the editable install directory or
         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
     :return: None
     """
     import flyte.config as config
@@ -228,6 +234,13 @@ async def init_from_config(path_or_config: str | Config | None = None, root_dir:
     cfg: config.Config
     if path_or_config is None or isinstance(path_or_config, str):
         # If a string is passed, treat it as a path to the config file
+        if path_or_config:
+            if not Path(path_or_config).exists():
+                raise InitializationError(
+                    "ConfigFileNotFoundError",
+                    "user",
+                    f"Configuration file '{path_or_config}' does not exist., current working directory is {Path.cwd()}",
+                )
         if root_dir and path_or_config:
             cfg = config.auto(str(root_dir / path_or_config))
         else:
@@ -251,6 +264,7 @@ async def init_from_config(path_or_config: str | Config | None = None, root_dir:
         client_id=cfg.platform.client_id,
         client_credentials_secret=cfg.platform.client_credentials_secret,
         root_dir=root_dir,
+        log_level=log_level,
     )
 
 

flyte/_internal/controllers/_local_controller.py

@@ -162,6 +162,7 @@ class LocalController:
                 action=action_id,
                 interface=_interface,
                 inputs_path=action_output_path,
+                name=_func.__name__,
             ),
             True,
         )
@@ -179,7 +180,7 @@ class LocalController:
 
         if info.interface.outputs and info.output:
             # If the result is not an AsyncGenerator, convert it directly
-            converted_outputs = await convert.convert_from_native_to_outputs(info.output, info.interface)
+            converted_outputs = await convert.convert_from_native_to_outputs(info.output, info.interface, info.name)
             assert converted_outputs
         elif info.error:
             # If there is an error, convert it to a native error

flyte/_internal/controllers/_trace.py

@@ -18,6 +18,7 @@ class TraceInfo:
     duration: Optional[timedelta] = None
     output: Optional[Any] = None
     error: Optional[Exception] = None
+    name: str = ""
 
     def add_outputs(self, output: Any, duration: timedelta):
         """

flyte/_internal/controllers/remote/_action.py

@@ -130,7 +130,7 @@ class Action:
         """
         from flyte._logging import logger
 
-        logger.info(f"In Action from_state {obj.action_id} {obj.phase} {obj.output_uri}")
+        logger.debug(f"In Action from_state {obj.action_id} {obj.phase} {obj.output_uri}")
         return cls(
             action_id=obj.action_id,
             parent_action_name=parent_action_name,

flyte/_internal/controllers/remote/_informer.py

@@ -235,7 +235,7 @@ class Informer:
                     await self._shared_queue.put(node)
                     # hack to work in the absence of sentinel
             except asyncio.CancelledError:
-                logger.warning(f"Watch cancelled: {self.name}")
+                logger.info(f"Watch cancelled: {self.name}")
                 return
             except asyncio.TimeoutError as e:
                 logger.error(f"Watch timeout: {self.name}", exc_info=e)

flyte/_internal/runtime/convert.py

@@ -11,7 +11,7 @@ import flyte.errors
 import flyte.storage as storage
 from flyte._protos.workflow import run_definition_pb2, task_definition_pb2
 from flyte.models import ActionID, NativeInterface, TaskContext
-from flyte.types import TypeEngine
+from flyte.types import TypeEngine, TypeTransformerFailedError
 
 
 @dataclass(frozen=True)
@@ -80,7 +80,7 @@ async def convert_from_native_to_inputs(interface: NativeInterface, *args, **kwa
     )
 
 
-async def convert_from_native_to_outputs(o: Any, interface: NativeInterface) -> Outputs:
+async def convert_from_native_to_outputs(o: Any, interface: NativeInterface, task_name: str = "") -> Outputs:
     # Always make it a tuple even if it's just one item to simplify logic below
     if not isinstance(o, tuple):
         o = (o,)
@@ -90,8 +90,11 @@ async def convert_from_native_to_outputs(o: Any, interface: NativeInterface) ->
     )
     named = []
     for (output_name, python_type), v in zip(interface.outputs.items(), o):
-        lit = await TypeEngine.to_literal(v, python_type, TypeEngine.to_literal_type(python_type))
-        named.append(run_definition_pb2.NamedLiteral(name=output_name, value=lit))
+        try:
+            lit = await TypeEngine.to_literal(v, python_type, TypeEngine.to_literal_type(python_type))
+            named.append(run_definition_pb2.NamedLiteral(name=output_name, value=lit))
+        except TypeTransformerFailedError as e:
+            raise flyte.errors.RuntimeDataValidationError(output_name, e, task_name)
 
     return Outputs(proto_outputs=run_definition_pb2.Outputs(literals=named))
 

flyte/_internal/runtime/task_serde.py

@@ -3,9 +3,11 @@ This module provides functionality to serialize and deserialize tasks to and fro
 It includes a Resolver interface for loading tasks, and functions to load classes and tasks.
 """
 
+import copy
 import importlib
+import typing
 from datetime import timedelta
-from typing import Optional, Type
+from typing import Optional, Type, cast
 
 from flyteidl.core import identifier_pb2, literals_pb2, security_pb2, tasks_pb2
 from google.protobuf import duration_pb2, wrappers_pb2
@@ -13,6 +15,7 @@ from google.protobuf import duration_pb2, wrappers_pb2
 import flyte.errors
 from flyte._cache.cache import VersionParameters, cache_from_request
 from flyte._logging import logger
+from flyte._pod import _PRIMARY_CONTAINER_NAME_FIELD, PodTemplate
 from flyte._protos.workflow import task_definition_pb2
 from flyte._secret import SecretRequest, secrets_from_request
 from flyte._task import AsyncFunctionTaskTemplate, TaskTemplate
@@ -121,17 +124,18 @@ def get_proto_task(task: TaskTemplate, serialize_context: SerializationContext)
     # if task.parent_env is None:
     # raise ValueError(f"Task {task.name} must have a parent environment")
 
-    #
-    # This pod will be incorrect when doing fast serialize
-    #
-    container = _get_urun_container(serialize_context, task)
+    # TODO Add support for SQL, extra_config, custom
+    extra_config: typing.Dict[str, str] = {}
+    custom = {}  # type: ignore
 
-    # TODO Add support for SQL, Pod, extra_config, custom
-    pod = None
     sql = None
-    # pod = task.get_k8s_pod(serialize_context)
-    extra_config = {}  # type: ignore
-    custom = {}  # type: ignore
+    if task.pod_template and not isinstance(task.pod_template, str):
+        container = None
+        pod = _get_k8s_pod(_get_urun_container(serialize_context, task), task.pod_template)
+        extra_config[_PRIMARY_CONTAINER_NAME_FIELD] = task.pod_template.primary_container_name
+    else:
+        container = _get_urun_container(serialize_context, task)
+        pod = None
 
     # -------------- CACHE HANDLING ----------------------
     task_cache = cache_from_request(task.cache)
@@ -210,6 +214,72 @@ def _get_urun_container(
     )
 
 
+def _sanitize_resource_name(resource: tasks_pb2.Resources.ResourceEntry) -> str:
+    return tasks_pb2.Resources.ResourceName.Name(resource.name).lower().replace("_", "-")
+
+
+def _get_k8s_pod(primary_container: tasks_pb2.Container, pod_template: PodTemplate) -> Optional[tasks_pb2.K8sPod]:
+    """
+    Get the K8sPod representation of the task template.
+    :param task: The task to convert.
+    :return: The K8sPod representation of the task template.
+    """
+    from kubernetes.client import ApiClient, V1PodSpec
+    from kubernetes.client.models import V1EnvVar, V1ResourceRequirements
+
+    pod_template = copy.deepcopy(pod_template)
+    containers = cast(V1PodSpec, pod_template.pod_spec).containers
+    primary_exists = False
+
+    for container in containers:
+        if container.name == pod_template.primary_container_name:
+            primary_exists = True
+            break
+
+    if not primary_exists:
+        raise ValueError(
+            "No primary container defined in the pod spec."
+            f" You must define a primary container with the name '{pod_template.primary_container_name}'."
+        )
+    final_containers = []
+
+    for container in containers:
+        # We overwrite the primary container attributes with the values given to ContainerTask.
+        # The attributes include: image, command, args, resource, and env (env is unioned)
+
+        if container.name == pod_template.primary_container_name:
+            if container.image is None:
+                # Copy the image from primary_container only if the image is not specified in the pod spec.
+                container.image = primary_container.image
+
+            container.command = list(primary_container.command)
+            container.args = list(primary_container.args)
+
+            limits, requests = {}, {}
+            for resource in primary_container.resources.limits:
+                limits[_sanitize_resource_name(resource)] = resource.value
+            for resource in primary_container.resources.requests:
+                requests[_sanitize_resource_name(resource)] = resource.value
+
+            resource_requirements = V1ResourceRequirements(limits=limits, requests=requests)
+            if len(limits) > 0 or len(requests) > 0:
+                # Important! Only copy over resource requirements if they are non-empty.
+                container.resources = resource_requirements
+
+            if primary_container.env is not None:
+                container.env = [V1EnvVar(name=e.key, value=e.value) for e in primary_container.env] + (
+                    container.env or []
+                )
+
+        final_containers.append(container)
+
+    cast(V1PodSpec, pod_template.pod_spec).containers = final_containers
+    pod_spec = ApiClient().sanitize_for_serialization(pod_template.pod_spec)
+
+    metadata = tasks_pb2.K8sObjectMetadata(labels=pod_template.labels, annotations=pod_template.annotations)
+    return tasks_pb2.K8sPod(pod_spec=pod_spec, metadata=metadata)
+
+
 def extract_code_bundle(task_spec: task_definition_pb2.TaskSpec) -> Optional[CodeBundle]:
     """
     Extract the code bundle from the task spec.

flyte/_internal/runtime/taskrunner.py

@@ -145,7 +145,7 @@ async def convert_and_run(
             return None, convert_from_native_to_error(err)
         if task.report:
             await flyte.report.flush.aio()
-        return await convert_from_native_to_outputs(out, task.native_interface), None
+        return await convert_from_native_to_outputs(out, task.native_interface, task.name), None
 
 
 async def extract_download_run_upload(

flyte/_logging.py

@@ -6,7 +6,7 @@ from typing import Optional
 
 from ._tools import ipython_check, is_in_cluster
 
-DEFAULT_LOG_LEVEL = logging.INFO
+DEFAULT_LOG_LEVEL = logging.WARNING
 
 
 def is_rich_logging_disabled() -> bool:

flyte/_pod.py

@@ -0,0 +1,19 @@
+from dataclasses import dataclass, field
+from typing import TYPE_CHECKING, Dict, Optional
+
+if TYPE_CHECKING:
+    from kubernetes.client import V1PodSpec
+
+
+_PRIMARY_CONTAINER_NAME_FIELD = "primary_container_name"
+_PRIMARY_CONTAINER_DEFAULT_NAME = "primary"
+
+
+@dataclass(init=True, repr=True, eq=True, frozen=False)
+class PodTemplate(object):
+    """Custom PodTemplate specification for a Task."""
+
+    pod_spec: Optional["V1PodSpec"] = field(default_factory=lambda: V1PodSpec())
+    primary_container_name: str = _PRIMARY_CONTAINER_DEFAULT_NAME
+    labels: Optional[Dict[str, str]] = None
+    annotations: Optional[Dict[str, str]] = None

flyte/_task.py

@@ -23,6 +23,7 @@ from typing import (
 
 from flyteidl.core.tasks_pb2 import DataLoadingConfig
 
+from flyte._pod import PodTemplate
 from flyte.errors import RuntimeSystemError, RuntimeUserError
 
 from ._cache import Cache, CacheRequest
@@ -37,8 +38,6 @@ from ._timeout import TimeoutType
 from .models import NativeInterface, SerializationContext
 
 if TYPE_CHECKING:
-    from kubernetes.client import V1PodTemplate
-
     from ._task_environment import TaskEnvironment
 
 P = ParamSpec("P")  # capture the function's parameters
@@ -96,8 +95,7 @@ class TaskTemplate(Generic[P, R]):
     env: Optional[Dict[str, str]] = None
     secrets: Optional[SecretRequest] = None
     timeout: Optional[TimeoutType] = None
-    primary_container_name: str = "primary"
-    pod_template: Optional[Union[str, V1PodTemplate]] = None
+    pod_template: Optional[Union[str, PodTemplate]] = None
     report: bool = False
 
     parent_env: Optional[weakref.ReferenceType[TaskEnvironment]] = None
@@ -108,15 +106,6 @@ class TaskTemplate(Generic[P, R]):
     _call_as_synchronous: bool = False
 
     def __post_init__(self):
-        # If pod_template is set to a pod, verify
-        if self.pod_template is not None and not isinstance(self.pod_template, str):
-            try:
-                from kubernetes.client import V1PodTemplate  # noqa: F401
-            except ImportError as e:
-                raise ImportError(
-                    "kubernetes is not installed, please install kubernetes package to use pod_template"
-                ) from e
-
         # Auto set the image based on the image request
         if self.image == "auto":
             self.image = Image.auto()

flyte/_utils/org_discovery.py

@@ -0,0 +1,31 @@
+def hostname_from_url(url: str) -> str:
+    """Parse a URL and return the hostname part."""
+
+    # Handle dns:/// format specifically (gRPC convention)
+    if url.startswith("dns:///"):
+        return url[7:]  # Skip the "dns:///" prefix
+
+    # Handle standard URL formats
+    import urllib.parse
+
+    parsed = urllib.parse.urlparse(url)
+    return parsed.netloc or parsed.path.lstrip("/").rsplit("/")[0]
+
+
+def org_from_endpoint(endpoint: str | None) -> str | None:
+    """
+    Extracts the organization from the endpoint URL. The organization is assumed to be the first part of the domain.
+    This is temporary until we have a proper organization discovery mechanism through APIs.
+
+    :param endpoint: The endpoint URL
+    :return: The organization name or None if not found
+    """
+    if not endpoint:
+        return None
+
+    hostname = hostname_from_url(endpoint)
+    domain_parts = hostname.split(".")
+    if len(domain_parts) > 2:
+        # Assuming the organization is the first part of the domain
+        return domain_parts[0]
+    return None

flyte/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.2.0b10'
-__version_tuple__ = version_tuple = (0, 2, 0, 'b10')
+__version__ = version = '0.2.0b11'
+__version_tuple__ = version_tuple = (0, 2, 0, 'b11')

flyte/cli/_common.py

@@ -78,7 +78,7 @@ class CLIConfig:
     log_level: int | None = logging.ERROR
     endpoint: str | None = None
     insecure: bool = False
-    org_override: str | None = None
+    org: str | None = None
 
     def replace(self, **kwargs) -> CLIConfig:
         """
@@ -90,7 +90,7 @@ class CLIConfig:
         from flyte.config._config import TaskConfig
 
         task_cfg = TaskConfig(
-            org=self.org_override or self.config.task.org,
+            org=self.org or self.config.task.org,
             project=project or self.config.task.project,
             domain=domain or self.config.task.domain,
         )

flyte/cli/_create.py

@@ -86,11 +86,9 @@ def secret(
     default=False,
     help="Force overwrite of the configuration file if it already exists.",
     show_default=True,
-    prompt="Are you sure you want to overwrite the configuration file?",
-    confirmation_prompt=True,
 )
 def config(
-    output: Path,
+    output: str,
     endpoint: str | None = None,
     insecure: bool = False,
     org: str | None = None,
@@ -105,8 +103,13 @@ def config(
     """
     import yaml
 
-    if output.exists() and not force:
-        raise click.BadParameter(f"Output file {output} already exists. Use --force to overwrite.")
+    output_path = Path(output)
+
+    if output_path.exists() and not force:
+        force = click.confirm(f"Overwrite [{output_path}]?", default=False)
+        if not force:
+            click.echo(f"Will not overwrite the existing config file at {output_path}")
+            return
 
     admin: Dict[str, Any] = {}
     if endpoint:
@@ -114,6 +117,11 @@ def config(
     if insecure:
         admin["insecure"] = insecure
 
+    if not org and endpoint:
+        from flyte._utils.org_discovery import org_from_endpoint
+
+        org = org_from_endpoint(endpoint)
+
     task: Dict[str, str] = {}
     if org:
         task["org"] = org
@@ -125,7 +133,7 @@ def config(
     if not admin and not task:
         raise click.BadParameter("At least one of --endpoint or --org must be provided.")
 
-    with open(output, "w") as f:
+    with open(output_path, "w") as f:
         d: Dict[str, Any] = {}
         if admin:
             d["admin"] = admin
@@ -133,4 +141,4 @@ def config(
             d["task"] = task
         yaml.dump(d, f)
 
-    click.echo(f"Config file created at {output}")
+    click.echo(f"Config file written to {output_path}")

flyte/cli/_params.py

@@ -110,7 +110,7 @@ class FileParamType(click.ParamType):
             p = pathlib.Path(value)
             if not p.exists() or not p.is_file():
                 raise click.BadParameter(f"parameter should be a valid file path, {value}")
-        return File(path=value)
+        return File.from_existing_remote(value)
 
 
 class PickleParamType(click.ParamType):

flyte/cli/_run.py

@@ -97,7 +97,7 @@ class RunTaskCommand(click.Command):
         if obj is None:
             import flyte.config
 
-            obj = CLIConfig(flyte.config.auto())
+            obj = CLIConfig(flyte.config.auto(), ctx)
 
         if not self.run_args.local:
             assert obj.endpoint, "CLI Config should have an endpoint"

flyte/cli/main.py

@@ -146,15 +146,11 @@ def main(
     cfg = config.auto(config_file=config_file)
     logger.debug(f"Using config file discovered at location {cfg.source}")
 
-    final_insecure = cfg.platform.insecure
-    if insecure is not None:
-        final_insecure = insecure
-
     ctx.obj = CLIConfig(
         log_level=log_level,
-        endpoint=endpoint or cfg.platform.endpoint,
-        insecure=final_insecure,
-        org_override=org or cfg.task.org,
+        endpoint=endpoint,
+        insecure=insecure,
+        org=org,
         config=cfg,
         ctx=ctx,
     )

flyte/errors.py

@@ -150,3 +150,14 @@ class LogsNotYetAvailableError(BaseRuntimeError):
 
     def __init__(self, message: str):
         super().__init__("LogsNotYetAvailable", "system", message, None)
+
+
+class RuntimeDataValidationError(RuntimeUserError):
+    """
+    This error is raised when the user tries to access a resource that does not exist or is invalid.
+    """
+
+    def __init__(self, var: str, e: Exception, task_name: str = ""):
+        super().__init__(
+            "DataValiationError", f"In task {task_name} variable {var}, failed to serialize/deserialize because {e}"
+        )

flyte/extras/_container.py

@@ -9,8 +9,6 @@ from flyte._logging import logger
 from flyte._task import TaskTemplate
 from flyte.models import NativeInterface, SerializationContext
 
-_PRIMARY_CONTAINER_NAME_FIELD = "primary_container_name"
-
 
 def _extract_command_key(cmd: str, **kwargs) -> List[Any] | None:
     """
@@ -263,8 +261,3 @@ class ContainerTask(TaskTemplate):
 
     def container_args(self, sctx: SerializationContext) -> List[str]:
         return self._cmd + (self._args if self._args else [])
-
-    def config(self, sctx: SerializationContext) -> Dict[str, str]:
-        if self.pod_template is None:
-            return {}
-        return {_PRIMARY_CONTAINER_NAME_FIELD: self.primary_container_name}

flyte/remote/_data.py

@@ -100,7 +100,8 @@ async def _upload_single_file(
             if put_resp.status_code != 200:
                 raise RuntimeSystemError(
                     "UploadFailed",
-                    f"Failed to upload {fp} to {resp.signed_url}, status code: {put_resp.status_code}",
+                    f"Failed to upload {fp} to {resp.signed_url}, status code: {put_resp.status_code}, "
+                    f"response: {put_resp.text}",
                 )
         # TODO in old code we did this
         #             if self._config.platform.insecure_skip_verify is True

flyte/types/_type_engine.py

@@ -1207,7 +1207,9 @@ class TypeEngine(typing.Generic[T]):
                 python_type = type_hints.get(k, type(d[k]))
                 e: BaseException = literal_map[k].exception()  # type: ignore
                 if isinstance(e, TypeError):
-                    raise TypeError(f"Error converting: {type(v)}, {python_type}, received_value {v}")
+                    raise TypeError(
+                        f"Error converting: Var:{k}, type:{type(v)}, into:{python_type}, received_value {v}"
+                    )
                 else:
                     raise e
             literal_map[k] = v.result()

{flyte-0.2.0b10.dist-info → flyte-0.2.0b11.dist-info}/METADATA

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

{flyte-0.2.0b10.dist-info → flyte-0.2.0b11.dist-info}/RECORD

@@ -1,4 +1,4 @@
-flyte/__init__.py,sha256=QzKbvypcEAJCF7g8oqpIZhpQfYOsqz6hEvT2OwK0IAc,1415
+flyte/__init__.py,sha256=e8-Obt_5OZDdC5Riznft0Y4ctQMYkYYO1UU7Z6EirLA,1464
 flyte/_build.py,sha256=MkgfLAPeL56YeVrGRNZUCZgbwzlEzVP3wLbl5Qru4yk,578
 flyte/_context.py,sha256=K0-TCt-_pHOoE5Xni87_8uIe2vCBOhfNQEtjGT4Hu4k,5239
 flyte/_deploy.py,sha256=NApDDNQRGauhZP_eBYUnVSQilooSLpmmHQNhqNa5jmk,8034
@@ -8,26 +8,27 @@ flyte/_environment.py,sha256=BkChtdVhWB3SwMSvetDZ-KiNBgRFlAXgq69PHT4zyG0,2942
 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/_initialize.py,sha256=N7q2n22iHSWmOB86ZLfgMd1En_XU6dz3OusB-t5nUwg,17073
+flyte/_initialize.py,sha256=oVLiidNRATEdDouzppaG2W_NYRok-YuNBjA7xlGpRAY,17696
 flyte/_interface.py,sha256=MP5o_qpIwfBNtAc7zo_cLSjMugsPyanuO6EgUSk4fBE,3644
-flyte/_logging.py,sha256=FQvF3W1kkFypbARcOQ7WZVXO0XJasXp8EhozF6E6-aQ,3379
+flyte/_logging.py,sha256=n3pi1cE_oqA03vQHlqSxexXcjRhCsZg1ODvUSoWlwk4,3382
 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=ePlFrABDR0ud_qxY55Nk4eATDialHGHsxNdTiVWYQkw,19919
 flyte/_secret.py,sha256=SqIHs6mi8hEkIIBZe3bI9jJsPt65Mt6dV5uh9_op1ME,2392
-flyte/_task.py,sha256=Gq34LbELz5dIc4dvYjgRXyzAnBvs8ED3iU1FWVuAFRE,18391
+flyte/_task.py,sha256=AHxg4lqHfExdSU6adwPiFzAT2EtrLI8mBdRxTUL1RgA,17902
 flyte/_task_environment.py,sha256=J1LFH9Zz1nPhlsrc_rYny1SS3QC1b55X7tRYoTG7Vk4,6815
 flyte/_timeout.py,sha256=zx5sFcbYmjJAJbZWSGzzX-BpC9HC7Jfs35T7vVhKwkk,1571
 flyte/_tools.py,sha256=JewkQZBR_M85tS6QY8e4xXue75jbOE48nID4ZHnc9jY,632
 flyte/_trace.py,sha256=7OQtQNosIlycTwaMjdc3GW4h3T3N0bYTsY6og4clPl8,5234
-flyte/_version.py,sha256=mlXpANdUuYfy0QPXXxhosF8MwMVypx0OlvhibSGO2P8,521
-flyte/errors.py,sha256=m2JUNqLC6anVW6UiDK_ihuA06q_Hkw1mIUMDskb2OW8,4289
+flyte/_version.py,sha256=D4lhYrgx59XZRRry4jkbv6B5ZVzqyxGy7BrlDWwfBR0,521
+flyte/errors.py,sha256=lJgSiZb2nZnuTZqBEB9rg-bV_GXVAYNjQFRuKWQskyY,4683
 flyte/models.py,sha256=A85HnTLqInJiMmQKhpl2IXb2Uh6_46tdRrwW2TTzD9o,12908
 flyte/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 flyte/_bin/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-flyte/_bin/runtime.py,sha256=Q1vajoQ913KUfCQSP0u8Qoj-AYgfvU2i2wI19ydqnDw,4849
+flyte/_bin/runtime.py,sha256=D6iFiVkOh3toKpBoJ_X9S1SeBDrEVGumSiqOrMhZG70,5032
 flyte/_cache/__init__.py,sha256=zhdO5UuHQRdzn8GHmSN40nrxfAmI4ihDRuHZM11U84Y,305
 flyte/_cache/cache.py,sha256=ErhWzzJdEjTIuEF4f-r6IBgko-3Al9iUs1Eq4O42TUE,5021
 flyte/_cache/defaults.py,sha256=gzJZW0QJPUfd2OPnGpv3tzIfwPtgFjAKoie3NP1P97U,217
@@ -39,14 +40,14 @@ flyte/_code_bundle/_utils.py,sha256=b0s3ZVKSRwaa_2CMTCqt2iRrUvTTW3FmlyqCD9k5BS0,
 flyte/_code_bundle/bundle.py,sha256=8T0gcXck6dmg-8L2-0G3B2iNjC-Xwydu806iyKneMMY,8789
 flyte/_internal/__init__.py,sha256=vjXgGzAAjy609YFkAy9_RVPuUlslsHSJBXCLNTVnqOY,136
 flyte/_internal/controllers/__init__.py,sha256=5CBnS9lb1VFMzZuRXUiaPhlN3G9qh7Aq9kTwxW5hsRw,4301
-flyte/_internal/controllers/_local_controller.py,sha256=LDwVxwm365L6-OEpwo5FcWe6abxK8L1231Hf54pYLZ8,7076
-flyte/_internal/controllers/_trace.py,sha256=Ga2b65sn9q2IoOwHBZV2inMYyO6-CSDwzN7E3pDxsEI,1126
+flyte/_internal/controllers/_local_controller.py,sha256=Wpgtd50C_ovIHpQSZC6asQc7iKyKIraEf-MAHCwcNJI,7124
+flyte/_internal/controllers/_trace.py,sha256=biI-lXSIe3gXuWI-KT6T-jTtojQCQ7BLOHTCG3J6MQc,1145
 flyte/_internal/controllers/remote/__init__.py,sha256=9_azH1eHLqY6VULpDugXi7Kf1kK1ODqEnsQ_3wM6IqU,1919
-flyte/_internal/controllers/remote/_action.py,sha256=w6vE1vPz1BwxvwfotDWjTNbDXfGEPrRBA8N3UVQ6P0w,4905
+flyte/_internal/controllers/remote/_action.py,sha256=5V26eE1ggtf95M8l_3wBRGrtbQ1DkOD_ePRT2bppMGM,4906
 flyte/_internal/controllers/remote/_client.py,sha256=HPbzbfaWZVv5wpOvKNtFXR6COiZDwd1cUJQqi60A7oU,1421
 flyte/_internal/controllers/remote/_controller.py,sha256=IbCRMTbQrNz96zjSZo2KHDmB0nW3dwT8VlWLu4zElGQ,19462
 flyte/_internal/controllers/remote/_core.py,sha256=2dka1rDnA8Ui_qhfE1ymZuN8E2BYQPn123h_eMixSiM,18091
-flyte/_internal/controllers/remote/_informer.py,sha256=JC35aHbEdwBN7cwDKbQj6koPUypTapYyK0wKxOBRBCo,14191
+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
@@ -57,12 +58,12 @@ flyte/_internal/resolvers/_task_module.py,sha256=jwy1QYygUK7xmpCZLt1SPTfJCkfox3C
 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=ThvyrR4EnuNu6i1rhgeKy259UMpbn_tmFY7YCBDNlh4,9122
+flyte/_internal/runtime/convert.py,sha256=0ttgaC8GrRkM2nXuG0K-7Kcg6nFsKWC9BG2LSEzgWzM,9325
 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=985eItmPsnA17CQdRXknjVDBK8wzOx4956AUuVjLsM8,9772
-flyte/_internal/runtime/taskrunner.py,sha256=74rnpABxEjqA7ZWxGZ1bXNU3rRT8Ba2utkuKXVpTwY4,7321
+flyte/_internal/runtime/task_serde.py,sha256=wN7lsusEUuEQE4-jPh0f_sTFZisH76o1VlMMXHC7etI,13012
+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
 flyte/_protos/common/authorization_pb2.py,sha256=6G7CAfq_Vq1qrm8JFkAnMAj0AaEipiX7MkjA7nk91-M,6707
@@ -134,25 +135,26 @@ flyte/_utils/coro_management.py,sha256=_JTt9x9fOc_1OSe03DSheYoKOvlonBB_4WNCS9XSa
 flyte/_utils/file_handling.py,sha256=iU4TxW--fCho_Eg5xTMODn96P03SxzF-V-5f-7bZAZY,2233
 flyte/_utils/helpers.py,sha256=45ZC2OSNKS66DkTvif8W8x7MH4KxluvAyn0a92mKRqM,4366
 flyte/_utils/lazy_module.py,sha256=fvXPjvZLzCfcI8Vzs4pKedUDdY0U_RQ1ZVrp9b8qBQY,1994
+flyte/_utils/org_discovery.py,sha256=80TAUg2VUGvUZtTOPlPpEYnNqJANdv_J-i41NjP9NrY,1055
 flyte/_utils/uv_script_parser.py,sha256=PxqD8lSMi6xv0uDd1s8LKB2IPZr4ttZJCUweqlyMTKk,1483
 flyte/cli/__init__.py,sha256=M02O-UGqQlA8JJ_jyWRiwQhTNc5CJJ7x9J7fNxTxBT0,52
 flyte/cli/_abort.py,sha256=lTftDmVXEIrFz1XAASCqWbXQEQDqRdTCJqY7izk2USI,593
-flyte/cli/_common.py,sha256=dwZ5OPz3x8-8uVtMiV1HpqA1MaoBJZH8yBON-axFSZs,10795
-flyte/cli/_create.py,sha256=DFdNemGqNVdlgDxcIqMmZJBF1zPtAHeWO1XeT2caEPs,4087
+flyte/cli/_common.py,sha256=CPGQlEvQZ8-MWEw9C610ti-s55HYvLYl313MFvI0eWY,10777
+flyte/cli/_create.py,sha256=GO_7qEOW1udDuHnZWTUMPqDyVLW_K2RKT-d3Qswpf6s,4273
 flyte/cli/_delete.py,sha256=qq5A9d6vXdYvYz-SECXiC6LU5rAzahNTZKiKacOtcZ4,545
 flyte/cli/_deploy.py,sha256=owGe_RVwC73p3dOPWYSo442OVuZyagF8LKTvC57xubQ,4466
 flyte/cli/_gen.py,sha256=vlE5l8UR1zz4RSdaRyUfYFvGR0TLxGcTYcP4dhA3Pvg,5458
 flyte/cli/_get.py,sha256=Pl9nHVHzRQT8IG3k_VAMJ2Xvaq9URsovm96op_153os,9843
-flyte/cli/_params.py,sha256=fs6fNBRGYXZMmjjcySi3U1O_pQNgCnW9ko89KCcS6nU,19450
-flyte/cli/_run.py,sha256=zRu4DTO__BVsLgpSLUbYR1Twvzc_2zivUhcviymCRf0,7884
-flyte/cli/main.py,sha256=PJUQ6slm1CcwIfeP6pbJKlPuyOqXxb_K8PjnZb8sbhw,4450
+flyte/cli/_params.py,sha256=cDeTvjOQP8EydVJUrncLeAxUaHvGorJyDvMSrAxapmM,19466
+flyte/cli/_run.py,sha256=dkuf6PfmizFU1-RWX_9TsTcGuUNDpjY7f6ok90nZ2iM,7889
+flyte/cli/main.py,sha256=P5a0Q4pKVKnc9QMdbdfp9I5ReGVykVCN-cUy0gURpLI,4287
 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=kMrYPmfnDFxPmFhI9cZeoBKzQ96jMCO4l4FpwV1b0PA,11202
+flyte/extras/_container.py,sha256=7fg43UScXevb9QdiyZLyf0xFrL5L3GkPvUWKQLl4R98,10939
 flyte/io/__init__.py,sha256=F7hlpin_1JJjsdFZSn7_jQgltPzsjETX1DCYGz-ELqI,629
 flyte/io/_dir.py,sha256=rih9CY1YjNX05bcAu5LG62Xoyij5GXAlv7jLyVF0je8,15310
 flyte/io/_file.py,sha256=kp5700SKPy5htmMhm4hE2ybb99Ykny1b0Kwm3huCWXs,15572
@@ -161,7 +163,7 @@ flyte/io/_structured_dataset/basic_dfs.py,sha256=77aYFrFigPC7cjM6UjCbU26REtXmwIB
 flyte/io/_structured_dataset/structured_dataset.py,sha256=ddRjz36RhNxIy0gakzdLStBzoo4cAOgXbNqiqt5YhMI,52645
 flyte/remote/__init__.py,sha256=zBWV88VF-L8430xVrOyk07EmLsOKhOUMVBsqFUDtO6Q,565
 flyte/remote/_console.py,sha256=avmELJPx8nQMAVPrHlh6jEIRPjrMwFpdZjJsWOOa9rE,660
-flyte/remote/_data.py,sha256=DPK85gB6M71RjxqIh1Q5PdZ9xcJ0m1w_3cT2lAO0r7w,5795
+flyte/remote/_data.py,sha256=B5nMF_o_ePRuG-V4IrTXctXXYn7uEJqUNs_5unX-2fo,5846
 flyte/remote/_logs.py,sha256=EOXg4OS8yYclsT6NASgOLMo0TA2sZpKb2MWZXpWBPuI,6404
 flyte/remote/_project.py,sha256=dTBYqORDAbLvh9WnPO1Ytuzw2vxNYZwwNsKE2_b0o14,2807
 flyte/remote/_run.py,sha256=9euHjYRX-xyxXuhn0MunYb9dmgl0FMU3a-FZNjJA4F8,31057
@@ -203,10 +205,10 @@ flyte/types/_interface.py,sha256=mY7mb8v2hJPGk7AU99gdOWl4_jArA1VFtjYGlE31SK0,953
 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=3PiRPYtE0UmEpTfwwcFWaiVmG1XOfXQl3Pj4v2zClNk,93602
+flyte/types/_type_engine.py,sha256=oX906WSQ-e8E1X1vwdMcqiQi2fUUFY9A4wVZZf2yoaw,93667
 flyte/types/_utils.py,sha256=pbts9E1_2LTdLygAY0UYTLYJ8AsN3BZyviSXvrtcutc,2626
-flyte-0.2.0b10.dist-info/METADATA,sha256=XALFwwiy3iCSCTVj80DPGhfL8gfPIVUOsyAMmbpNzac,4811
-flyte-0.2.0b10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-flyte-0.2.0b10.dist-info/entry_points.txt,sha256=MIq2z5dBurdCJfpXfMKzgBv7sJOakKRYxr8G0cMiTrg,75
-flyte-0.2.0b10.dist-info/top_level.txt,sha256=7dkyFbikvA12LEZEqawx8oDG1CMod6hTliPj7iWzgYo,6
-flyte-0.2.0b10.dist-info/RECORD,,
+flyte-0.2.0b11.dist-info/METADATA,sha256=a3P-yCWvVVslOcR357eSJVKR4WzzebAZCQKcinxd78A,4811
+flyte-0.2.0b11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+flyte-0.2.0b11.dist-info/entry_points.txt,sha256=MIq2z5dBurdCJfpXfMKzgBv7sJOakKRYxr8G0cMiTrg,75
+flyte-0.2.0b11.dist-info/top_level.txt,sha256=7dkyFbikvA12LEZEqawx8oDG1CMod6hTliPj7iWzgYo,6
+flyte-0.2.0b11.dist-info/RECORD,,