flyte 0.0.1b1__py3-none-any.whl → 0.0.1b2__py3-none-any.whl

flyte/_cli/_common.py

@@ -12,13 +12,10 @@ from typing import Any, Dict, Iterable, List, Optional
 
 import rich.box
 import rich_click as click
-from rich.console import Console
 from rich.panel import Panel
 from rich.table import Table
-from rich.traceback import Traceback
 
 import flyte.errors
-from flyte.config import Config
 
 PREFERRED_BORDER_COLOR = "dim cyan"
 PREFERRED_ACCENT_COLOR = "bold #FFD700"
@@ -73,7 +70,6 @@ class CLIConfig:
     endpoint: str | None = None
     insecure: bool = False
     org_override: str | None = None
-    config: Config | None = None
 
     def replace(self, **kwargs) -> CLIConfig:
         """
@@ -91,7 +87,6 @@ class CLIConfig:
             project=project,
             domain=domain,
             log_level=self.log_level,
-            config=self.config,
         )
 
 
@@ -116,14 +111,7 @@ class InvokeBaseMixin:
             raise click.ClickException(f"RPC error invoking command: {e!s}") from e
         except flyte.errors.InitializationError:
             raise click.ClickException("Initialize the CLI with a remote configuration. For example, pass --endpoint")
-        except click.exceptions.Exit as e:
-            # This is a normal exit, do nothing
-            raise e
         except Exception as e:
-            if ctx.obj and ctx.obj.log_level and ctx.obj.log_level <= logging.DEBUG:
-                # If the user has requested verbose output, print the full traceback
-                console = Console()
-                console.print(Traceback.from_exception(type(e), e, e.__traceback__))
             raise click.ClickException(f"Error invoking command: {e}") from e
 
 

flyte/_cli/_run.py

@@ -1,13 +1,12 @@
 from __future__ import annotations
 
-import inspect
 from dataclasses import dataclass, field, fields
 from pathlib import Path
 from types import ModuleType
 from typing import Any, Dict, List, cast
 
 import click
-from click import Context, Parameter
+from click import Context
 from rich.console import Console
 from typing_extensions import get_args
 
@@ -18,7 +17,6 @@ from .._task import TaskTemplate
 from ..remote import Run
 from . import _common as common
 from ._common import CLIConfig
-from ._params import to_click_option
 
 
 @dataclass
@@ -79,7 +77,7 @@ class RunTaskCommand(click.Command):
             copy_style=self.run_args.copy_style,
             version=self.run_args.copy_style,
             mode="local" if self.run_args.local else "remote",
-        ).run(self.obj, **ctx.params)
+        ).run(self.obj)
         if isinstance(r, Run) and r.action is not None:
             console = Console()
             console.print(
@@ -91,26 +89,6 @@ class RunTaskCommand(click.Command):
                 )
             )
 
-    def get_params(self, ctx: Context) -> List[Parameter]:
-        # Note this function may be called multiple times by click.
-        task = self.obj
-        from .._internal.runtime.types_serde import transform_native_to_typed_interface
-
-        interface = transform_native_to_typed_interface(task.native_interface)
-        if interface is None:
-            return super().get_params(ctx)
-        inputs_interface = task.native_interface.inputs
-
-        params: List[Parameter] = []
-        for name, var in interface.inputs.variables.items():
-            default_val = None
-            if inputs_interface[name][1] is not inspect._empty:
-                default_val = inputs_interface[name][1]
-            params.append(to_click_option(name, var, inputs_interface[name][0], default_val))
-
-        self.params = params
-        return super().get_params(ctx)
-
 
 class TaskPerFileGroup(common.ObjectsPerFileGroup):
     """

flyte/_cli/main.py

@@ -1,6 +1,5 @@
 import rich_click as click
 
-from ..config import Config
 from ._common import CLIConfig
 from ._create import create
 from ._deploy import deploy
@@ -58,38 +57,13 @@ def _verbosity_to_loglevel(verbosity: int) -> int | None:
     required=False,
     help="Override for org",
 )
-@click.option(
-    "-c",
-    "--config",
-    "config_file",
-    required=False,
-    type=click.Path(exists=True),
-    help="Path to config file (YAML format) to use for the CLI. If not specified,"
-    " the default config file will be used.",
-)
 @click.pass_context
-def main(
-    ctx: click.Context,
-    endpoint: str | None,
-    insecure: bool,
-    verbose: int,
-    org_override: str | None,
-    config_file: str | None,
-):
+def main(ctx: click.Context, endpoint: str | None, insecure: bool, verbose: int, org_override: str | None):
     """
     v2 cli. Root command, please use one of the subcommands.
     """
     log_level = _verbosity_to_loglevel(verbose)
-
-    config = Config.auto(config_file=config_file)
-
-    ctx.obj = CLIConfig(
-        log_level=log_level,
-        endpoint=endpoint or config.platform.endpoint,
-        insecure=insecure or config.platform.insecure,
-        org_override=org_override or config.task.org,
-        config=config,
-    )
+    ctx.obj = CLIConfig(log_level=log_level, endpoint=endpoint, insecure=insecure, org_override=org_override)
 
 
 main.add_command(run)

flyte/_image.py

@@ -291,17 +291,13 @@ class Image:
     def _get_default_image_for(cls, python_version: Tuple[int, int], flyte_version: Optional[str] = None) -> Image:
         # Would love a way to move this outside of this class (but still needs to be accessible via Image.auto())
         # this default image definition may need to be updated once there is a released pypi version
-        preset_tag = None
-        dev_mode = False
+        from flyte._version import __version__
+
+        dev_mode = cls._is_editable_install() or (__version__ and "dev" in __version__)
         if flyte_version is None:
-            from flyte._version import __version__
-            if __version__ and "dev" in __version__:
-                dev_mode = True
-                flyte_version = __version__.replace("+", "-")
-
-        if flyte_version:
-            preset_tag = flyte_version if flyte_version.startswith("v") else f"v{flyte_version}"
-            preset_tag = f"py{python_version[0]}.{python_version[1]}-{preset_tag}"
+            flyte_version = __version__.replace("+", "-")
+        preset_tag = flyte_version if flyte_version.startswith("v") else f"v{flyte_version}"
+        preset_tag = f"py{python_version[0]}.{python_version[1]}-{preset_tag}"
         image = Image(
             base_image=f"python:{python_version[0]}.{python_version[1]}-slim-bookworm",
             registry=_BASE_REGISTRY,

flyte/_initialize.py

@@ -17,7 +17,6 @@ from ._logging import initialize_logger
 from ._tools import ipython_check
 
 if TYPE_CHECKING:
-    from flyte.config import Config
     from flyte.remote._client.auth import AuthType, ClientConfig
     from flyte.remote._client.controlplane import ClientSet
 
@@ -346,7 +345,6 @@ async def init(
     rpc_retries: int = 3,
     http_proxy_url: str | None = None,
     storage: Storage | None = None,
-    config: Config | None = None,
 ) -> None:
     """
     Initialize the Flyte system with the given configuration. This method should be called before any other Flyte
@@ -380,7 +378,6 @@ async def init(
     :param insecure: insecure flag for the client
     :param storage: Optional blob store (S3, GCS, Azure) configuration if needed to access (i.e. using Minio)
     :param org: Optional organization override for the client. Should be set by auth instead.
-    :param config: Optional config to override the init parameters
 
     :return: None
     """
@@ -395,39 +392,33 @@ async def init(
     global _init_config  # noqa: PLW0603
 
     with _init_lock:
-        if config is None:
-            from flyte.config import Config
-
-            config = Config.auto()
-        platform_cfg = config.platform
-        task_cfg = config.task
         client = None
-        if endpoint or platform_cfg.endpoint or api_key:
+        if endpoint or api_key:
             client = await _initialize_client(
                 api_key=api_key,
-                auth_type=auth_type or platform_cfg.auth_mode,
-                endpoint=endpoint or platform_cfg.endpoint,
+                auth_type=auth_type,
+                endpoint=endpoint,
                 headless=headless,
-                insecure=insecure or platform_cfg.insecure,
-                insecure_skip_verify=insecure_skip_verify or platform_cfg.insecure_skip_verify,
-                ca_cert_file_path=ca_cert_file_path or platform_cfg.ca_cert_file_path,
-                command=command or platform_cfg.command,
-                proxy_command=proxy_command or platform_cfg.proxy_command,
-                client_id=client_id or platform_cfg.client_id,
-                client_credentials_secret=client_credentials_secret or platform_cfg.client_credentials_secret,
+                insecure=insecure,
+                insecure_skip_verify=insecure_skip_verify,
+                ca_cert_file_path=ca_cert_file_path,
+                command=command,
+                proxy_command=proxy_command,
+                client_id=client_id,
+                client_credentials_secret=client_credentials_secret,
                 client_config=auth_client_config,
-                rpc_retries=rpc_retries or platform_cfg.rpc_retries,
-                http_proxy_url=http_proxy_url or platform_cfg.http_proxy_url,
+                rpc_retries=rpc_retries,
+                http_proxy_url=http_proxy_url,
             )
 
         root_dir = root_dir or get_cwd_editable_install() or Path.cwd()
         _init_config = _InitConfig(
             root_dir=root_dir,
-            project=project or task_cfg.project,
-            domain=domain or task_cfg.domain,
+            project=project,
+            domain=domain,
             client=client,
             storage=storage,
-            org=org or task_cfg.org,
+            org=org,
         )
 
 

flyte/_internal/imagebuild/docker_builder.py

@@ -43,7 +43,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 \
-    uv pip install --python $$UV_PYTHON $PIP_INSTALL_ARGS
+    uv pip install --prerelease=allow --python $$UV_PYTHON $PIP_INSTALL_ARGS
 """)
 
 APT_INSTALL_COMMAND_TEMPLATE = Template("""\
@@ -54,7 +54,7 @@ RUN --mount=type=cache,sharing=locked,mode=0777,target=/var/cache/apt,id=apt \
 
 UV_PYTHON_INSTALL_COMMAND = Template("""\
 RUN --mount=type=cache,sharing=locked,mode=0777,target=/root/.cache/uv,id=uv \
-    uv pip install $PIP_INSTALL_ARGS
+    uv pip install --prerelease=allow $PIP_INSTALL_ARGS
 """)
 
 # uv pip install --python /root/env/bin/python

flyte/_internal/runtime/convert.py

@@ -60,12 +60,6 @@ async def convert_from_native_to_inputs(interface: NativeInterface, *args, **kwa
     kwargs = interface.convert_to_kwargs(*args, **kwargs)
     if len(kwargs) == 0:
         return Inputs.empty()
-    # fill in defaults if missing
-    for input_name, (input_type, default_value) in interface.inputs.items():
-        if input_name not in kwargs:
-            if default_value is not None:
-                kwargs[input_name] = default_value
-            # todo: fill in Nones for optional inputs
     if len(kwargs) < len(interface.inputs):
         raise ValueError(
             f"Received {len(kwargs)} inputs but interface has {len(interface.inputs)}. "

flyte/_run.py

@@ -277,6 +277,7 @@ class _Runner:
         from flyte._internal.runtime.entrypoints import direct_dispatch
 
         controller = create_controller(ct="local")
+
         inputs = await convert_from_native_to_inputs(obj.native_interface, *args, **kwargs)
         if self._name is None:
             action = ActionID.create_random()

flyte/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.0.1b1'
-__version_tuple__ = version_tuple = (0, 0, 1, 'b1')
+__version__ = version = '0.0.1b2'
+__version_tuple__ = version_tuple = (0, 0, 1, 'b2')

flyte/remote/_console.py

@@ -7,7 +7,7 @@ def _get_http_domain(endpoint: str, insecure: bool) -> str:
     if parsed.scheme == "dns":
         domain = parsed.path.lstrip("/")
     else:
-        domain = parsed.netloc or parsed.path
+        domain = parsed.netloc
     # TODO: make console url configurable
     if domain.split(":")[0] == "localhost":
         domain = "localhost:8080"

flyte/types/_type_engine.py

@@ -450,6 +450,7 @@ class DataclassTransformer(TypeTransformer[object]):
         # However, FooSchema is created by flytekit and it's not equal to the user-defined dataclass (Foo).
         # Therefore, we should iterate all attributes in the dataclass and check the type of value in dataclass
         #   matches the expected_type.
+
         expected_fields_dict = {}
 
         for f in dataclasses.fields(expected_type):
@@ -1991,9 +1992,7 @@ DatetimeTransformer = SimpleTransformer(
     datetime.datetime,
     types_pb2.LiteralType(simple=types_pb2.SimpleType.DATETIME),
     lambda x: Literal(scalar=Scalar(primitive=Primitive(datetime=x))),
-    lambda x: x.scalar.primitive.datetime.ToDatetime().replace(tzinfo=datetime.timezone.utc)
-    if x.scalar.primitive.HasField("datetime")
-    else None,
+    lambda x: x.scalar.primitive.datetime if x.scalar.primitive.HasField("datetime") else None,
 )
 
 TimedeltaTransformer = SimpleTransformer(
@@ -2001,7 +2000,7 @@ TimedeltaTransformer = SimpleTransformer(
     datetime.timedelta,
     types_pb2.LiteralType(simple=types_pb2.SimpleType.DURATION),
     lambda x: Literal(scalar=Scalar(primitive=Primitive(duration=x))),
-    lambda x: x.scalar.primitive.duration.ToTimedelta() if x.scalar.primitive.HasField("duration") else None,
+    lambda x: x.scalar.primitive.duration if x.scalar.primitive.HasField("duration") else None,
 )
 
 DateTransformer = SimpleTransformer(

{flyte-0.0.1b1.dist-info → flyte-0.0.1b2.dist-info}/METADATA

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