flyte 0.2.0b21__py3-none-any.whl → 0.2.0b23__py3-none-any.whl

flyte/_image.py

@@ -10,6 +10,7 @@ from pathlib import Path
 from typing import Callable, ClassVar, Dict, List, Literal, Optional, Tuple, TypeVar, Union
 
 import rich.repr
+from packaging.version import Version
 
 # Supported Python versions
 PYTHON_3_10 = (3, 10)
@@ -366,7 +367,6 @@ class Image:
             base_image=f"python:{python_version[0]}.{python_version[1]}-slim-bookworm",
             registry=_BASE_REGISTRY,
             name=_DEFAULT_IMAGE_NAME,
-            # _tag=preset_tag,
             platform=("linux/amd64", "linux/arm64"),
         )
         labels_and_user = _DockerLines(
@@ -393,7 +393,10 @@ class Image:
         if dev_mode:
             image = image._with_local_v2()
         else:
-            image = image.with_pip_packages(f"flyte=={flyte_version}")
+            if Version(flyte_version).is_devrelease or Version(flyte_version).is_prerelease:
+                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)
         # 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.

flyte/_internal/imagebuild/docker_builder.py

@@ -72,7 +72,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 --prerelease=allow $PIP_INSTALL_ARGS
+    uv pip install $PIP_INSTALL_ARGS
 """)
 
 # uv pip install --python /root/env/bin/python

flyte/_internal/runtime/task_serde.py

@@ -222,7 +222,13 @@ def _get_urun_container(
         img_uri = task_template.image.uri
     elif image_id not in serialize_context.image_cache.image_lookup:
         img_uri = task_template.image.uri
-        logger.warning(f"Image uri {img_uri} not found in image_cache {serialize_context.image_cache.image_lookup}")
+        from flyte._version import __version__
+        logger.warning(
+            f"Image {task_template.image} not found in the image cache: {serialize_context.image_cache.image_lookup}.\n"
+            f"This typically occurs when the Flyte SDK version (`{__version__}`) used in the task environment "
+            f"differs from the version used to compile or deploy it.\n"
+            f"Ensure both environments use the same Flyte SDK version to avoid inconsistencies in image resolution."
+        )
     else:
         img_uri = serialize_context.image_cache.image_lookup[image_id]
 

flyte/_version.py

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

flyte/cli/_common.py

@@ -151,7 +151,9 @@ class InvokeBaseMixin:
                 # 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
+                exit(1)
+            else:
+                raise click.ClickException(f"Error invoking command: {e}") from e
 
 
 class CommandBase(InvokeBaseMixin, click.RichCommand):

{flyte-0.2.0b21.dist-info → flyte-0.2.0b23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flyte
-Version: 0.2.0b21
+Version: 0.2.0b23
 Summary: Add your description here
 Author-email: Ketan Umare <kumare3@users.noreply.github.com>
 Requires-Python: >=3.10
@@ -16,7 +16,7 @@ Requires-Dist: protobuf>=6.30.1
 Requires-Dist: pydantic>=2.10.6
 Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: rich-click>=1.8.9
-Requires-Dist: httpx>=0.28.1
+Requires-Dist: httpx<1.0.0,>=0.28.1
 Requires-Dist: keyring>=25.6.0
 Requires-Dist: msgpack>=1.1.0
 Requires-Dist: toml>=0.10.2

{flyte-0.2.0b21.dist-info → flyte-0.2.0b23.dist-info}/RECORD

@@ -8,7 +8,7 @@ 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=6d_bxy2KOn0_wj895iPOxXKN5uzygUeT2fOhxG95dg4,30633
+flyte/_image.py,sha256=_KYygxYcS9eR8oal79clyAEmqcnlh0XBgCXbo4Aoktc,30840
 flyte/_initialize.py,sha256=xKl_LYMluRt21wWqa6RTKuLo0_DCbSaTfUk27_brtNk,18232
 flyte/_interface.py,sha256=1B9zIwFDjiVp_3l_mk8EpA4g3Re-6DUBEBi9z9vDvPs,3504
 flyte/_logging.py,sha256=_yNo-Nx2yzh0MLoZGbnIYHGKei4wtQmSGM0lE30Ev7w,3662
@@ -24,7 +24,7 @@ flyte/_task_environment.py,sha256=B493b54xpnNFNtHek3cmT4XuNiPNVf2gx9Yylyb1ncw,69
 flyte/_timeout.py,sha256=zx5sFcbYmjJAJbZWSGzzX-BpC9HC7Jfs35T7vVhKwkk,1571
 flyte/_tools.py,sha256=JewkQZBR_M85tS6QY8e4xXue75jbOE48nID4ZHnc9jY,632
 flyte/_trace.py,sha256=C788bgoSc3st8kE8Cae2xegnLx2CT6uuRKKfaDrDUys,5122
-flyte/_version.py,sha256=sbSySP-E_nMvVuSaQWbQZTGdTG_B1w6UKrkfQnFlsN4,521
+flyte/_version.py,sha256=-0paj9rX04GbPr33SZJMHLec76ZpD2KW8H6GxHKP0Pg,521
 flyte/errors.py,sha256=skXcdexLisFZFcTnUmMvMmuh4Ty96oJkyLKaipzkyeI,4954
 flyte/models.py,sha256=my7Vxo-NK6gHGahyqtHr42wYjPGw0nl2SGBBoSb6hIc,14733
 flyte/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -51,7 +51,7 @@ flyte/_internal/controllers/remote/_core.py,sha256=UKjiL3dGidXcR1dPzJswn67g38HQm
 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=4fPz5RN3QB1DGw5O2sZo7x71F3EWhyE1jmCCqy4iezk,573
-flyte/_internal/imagebuild/docker_builder.py,sha256=LIfqtLQ4NVxkjgWlASiIi_xqvxhztlUDTVt4P7McLDo,15946
+flyte/_internal/imagebuild/docker_builder.py,sha256=4-AJLX9CgPmVblzV-caIn3uvHLIlBMntOUKUmC-ptM8,15927
 flyte/_internal/imagebuild/image_builder.py,sha256=dXBXl62qcPabus6dR3eP8P9mBGNhpZHZ2Xm12AymKkk,11150
 flyte/_internal/imagebuild/remote_builder.py,sha256=wm-D3xhdrk8OSQxeyI0nfMpw3P1gW7mqDKAOKOnwY6o,10089
 flyte/_internal/resolvers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -63,7 +63,7 @@ flyte/_internal/runtime/convert.py,sha256=rvKXEDQxFLyAslSJeOrqeG1kkXaYxHWxkBMloa
 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=TObMVsSjVcQhcY8-nY81pbvrz7TP-adDD5xV-LqAaxM,4813
-flyte/_internal/runtime/task_serde.py,sha256=wjYZSr0aXCvo6egWLlTrrRgYbx-V6L1clY2jstJNQlk,14039
+flyte/_internal/runtime/task_serde.py,sha256=e_EMZIvcsqmY7f7RSmXmUjFhHefclxYd_P6LVknj0Qk,14425
 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
@@ -153,7 +153,7 @@ flyte/_utils/uv_script_parser.py,sha256=PxqD8lSMi6xv0uDd1s8LKB2IPZr4ttZJCUweqlyM
 flyte/cli/__init__.py,sha256=aeCcumeP9xD_5aCmaRYUPCe2QRJSGCaxcUbTZ3co768,341
 flyte/cli/_abort.py,sha256=Ty-63Gtd2PUn6lCuL5AaasfBoPu7TDSU5EQKVbkF4qw,661
 flyte/cli/_build.py,sha256=SBgybTVWOZ22VBHFL8CVFB_oo34lF9wvlwNirYFFyk0,3543
-flyte/cli/_common.py,sha256=CFBmdc0c59XqWb7PuZfLr7Z55HXnlqKaxDCq1sgY9MU,11115
+flyte/cli/_common.py,sha256=prQPs8_qSNCvhBPgHa6AY8c9OwutWw3js9jd5vdL_kk,11161
 flyte/cli/_create.py,sha256=8U4kePYmfW02j54184IurNYiTGvjdr3J-ERQM9goHdg,4715
 flyte/cli/_delete.py,sha256=VTmXv09PBjkdtyl23mbSjIQQlN7Y1AI_bO0GkHP-f9E,546
 flyte/cli/_deploy.py,sha256=Zxm7vn1zrqmll73CJTiVGYJI95P-XBI1AlhOlmbmkD0,4635
@@ -221,8 +221,8 @@ 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=CCjpqXNX2BMza2cKq42hJXwabWy8GWsimsgiGZJ_00M,96583
 flyte/types/_utils.py,sha256=pbts9E1_2LTdLygAY0UYTLYJ8AsN3BZyviSXvrtcutc,2626
-flyte-0.2.0b21.dist-info/METADATA,sha256=_A_r-O2ibG-Y8OBOhv9xiPuUXGqQyWtghF5ZY5NzLRo,5850
-flyte-0.2.0b21.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-flyte-0.2.0b21.dist-info/entry_points.txt,sha256=MIq2z5dBurdCJfpXfMKzgBv7sJOakKRYxr8G0cMiTrg,75
-flyte-0.2.0b21.dist-info/top_level.txt,sha256=7dkyFbikvA12LEZEqawx8oDG1CMod6hTliPj7iWzgYo,6
-flyte-0.2.0b21.dist-info/RECORD,,
+flyte-0.2.0b23.dist-info/METADATA,sha256=40x3MdlqXmAjoqEIVKPghhlV4xoHK27HIUrZPrz2Gdc,5857
+flyte-0.2.0b23.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+flyte-0.2.0b23.dist-info/entry_points.txt,sha256=MIq2z5dBurdCJfpXfMKzgBv7sJOakKRYxr8G0cMiTrg,75
+flyte-0.2.0b23.dist-info/top_level.txt,sha256=7dkyFbikvA12LEZEqawx8oDG1CMod6hTliPj7iWzgYo,6
+flyte-0.2.0b23.dist-info/RECORD,,