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

flyte/_image.py

@@ -291,10 +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
-        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}"
+        from flyte._version import __version__
+
+        dev_mode = cls._is_editable_install() or (__version__ and "dev" in __version__)
+        if flyte_version is None:
+            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,
@@ -325,18 +328,12 @@ class Image:
         base_packages = ["kubernetes", "msgpack", "mashumaro"]
 
         # Add in flyte library
-        if flyte_version:
-            base_packages.append(f"flyte=={flyte_version}")
+        if dev_mode:
             image = image.with_pip_packages(base_packages)
+            image = image.with_local_v2()
         else:
-            from flyte._version import __version__
-
-            if cls._is_editable_install() or (__version__ and "dev" in __version__):
-                image = image.with_pip_packages(base_packages)
-                image = image.with_local_v2()
-            else:
-                base_packages.append(f"flyte=={__version__}")
-                image = image.with_pip_packages(base_packages)
+            base_packages.append(f"flyte=={flyte_version}")
+            image = image.with_pip_packages(base_packages)
 
         return image
 

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/_version.py

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

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

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

{flyte-0.0.1b0.dist-info → flyte-0.0.1b2.dist-info}/RECORD

@@ -9,7 +9,7 @@ flyte/_docstring.py,sha256=SsG0Ab_YMAwy2ABJlEo3eBKlyC3kwPdnDJ1FIms-ZBQ,1127
 flyte/_environment.py,sha256=ft0EsyFg6OnoIFPFbwkABLcq676veIH3TTR4SNilrj8,1499
 flyte/_group.py,sha256=YmkTWJlO39wrrcOeNV3QUQUriNRwtdMFE_r8sXLLpSI,752
 flyte/_hash.py,sha256=Of_Zl_DzzzF2jp4ZsLm-3o-xJFCCJ8_GubmLI1htx78,504
-flyte/_image.py,sha256=fiOib4dPYi5WQXGXJ2t--EGNupF9QlZxb6VDZWzItRE,29193
+flyte/_image.py,sha256=8xEGmAALY6jQAsLfJQH9NweeVUaSTWivFEQt-JchN24,29068
 flyte/_initialize.py,sha256=Bli8Kg-RnEgYfCnjjOx6c9TNc2uhR_RsCc7AWmPfq1k,22595
 flyte/_interface.py,sha256=MP5o_qpIwfBNtAc7zo_cLSjMugsPyanuO6EgUSk4fBE,3644
 flyte/_logging.py,sha256=FQvF3W1kkFypbARcOQ7WZVXO0XJasXp8EhozF6E6-aQ,3379
@@ -23,7 +23,7 @@ flyte/_task_environment.py,sha256=1JSHkCvS-Ereh389mu1Z2eNIqsTXQGV-ZLiiqW09RnY,84
 flyte/_timeout.py,sha256=zx5sFcbYmjJAJbZWSGzzX-BpC9HC7Jfs35T7vVhKwkk,1571
 flyte/_tools.py,sha256=JewkQZBR_M85tS6QY8e4xXue75jbOE48nID4ZHnc9jY,632
 flyte/_trace.py,sha256=jWJqmIntfubetcS4S705tIyoGR4jnXaPgS3JuK4gkuo,5502
-flyte/_version.py,sha256=r2ohyOLVZBZTuwQr4jbH-dtebjgnKkEbRiB3h5hRT54,519
+flyte/_version.py,sha256=ZBnd207H0IMjqbSR-0BjJfRZDgicaLjViz-Z01imWKA,519
 flyte/errors.py,sha256=ulVzQ3TncddYOqQ3gvauDJAHTShWTpY0XZpqNqBZyOY,4036
 flyte/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 flyte/_bin/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -58,7 +58,7 @@ flyte/_internal/controllers/remote/_core.py,sha256=NRPfL7PO8UKywr4lyBqSeplqWJpXH
 flyte/_internal/controllers/remote/_informer.py,sha256=6WPaT1EmDcIwQ3VlujGWICzHy-kaGhMut_zBh2ShZnE,14186
 flyte/_internal/controllers/remote/_service_protocol.py,sha256=-p5FXdo5pp5xmr9fJ0sGB9aE7mimeC6x13bqwyIR77k,1345
 flyte/_internal/imagebuild/__init__.py,sha256=cLXVxkAyFpbdC1y-k3Rb6FRW9f_xpoRQWVn__G9IqKs,354
-flyte/_internal/imagebuild/docker_builder.py,sha256=xkfs22WGskEAzxmyEc-ZJ4m5ogLGBCSJoxdWXoHC2tU,13948
+flyte/_internal/imagebuild/docker_builder.py,sha256=bkr2fs9jInTq8jqU8ka7NGvp0RPfYhbTfX1RqtqTvvs,13986
 flyte/_internal/imagebuild/image_builder.py,sha256=q-TRcF_t2nS8FqBOtIIdnSo1Jrq378jqzceYDPzne8Q,9902
 flyte/_internal/imagebuild/remote_builder.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 flyte/_internal/resolvers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -383,8 +383,8 @@ union/types/_renderer.py,sha256=_ERSeK52wUXH72sb6aKLJxByP1nZSvuZiV0LQdOm4kI,4818
 union/types/_string_literals.py,sha256=zsqUg5TP3s5wheW-QTE48EbafUWEdlAHWqK-49PaQZo,4231
 union/types/_type_engine.py,sha256=KQ5d7JSASrOeIXwBxASiTEXJB5T59ZuYRgndr_kxUvA,90895
 union/types/_utils.py,sha256=pbts9E1_2LTdLygAY0UYTLYJ8AsN3BZyviSXvrtcutc,2626
-flyte-0.0.1b0.dist-info/METADATA,sha256=KicPQ05gSucJRod75iilTRAu3K6uqWpBMX_GiVb--vk,10279
-flyte-0.0.1b0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-flyte-0.0.1b0.dist-info/entry_points.txt,sha256=xitFzPlyODadzpIwr-x1mFIpz0IFKpMUJl3dnmxgyPc,76
-flyte-0.0.1b0.dist-info/top_level.txt,sha256=7dkyFbikvA12LEZEqawx8oDG1CMod6hTliPj7iWzgYo,6
-flyte-0.0.1b0.dist-info/RECORD,,
+flyte-0.0.1b2.dist-info/METADATA,sha256=aIhvPgqB-ggx4ZGhEOy5s-Yu5XdC6FoIk31SBvko5MA,10279
+flyte-0.0.1b2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+flyte-0.0.1b2.dist-info/entry_points.txt,sha256=xitFzPlyODadzpIwr-x1mFIpz0IFKpMUJl3dnmxgyPc,76
+flyte-0.0.1b2.dist-info/top_level.txt,sha256=7dkyFbikvA12LEZEqawx8oDG1CMod6hTliPj7iWzgYo,6
+flyte-0.0.1b2.dist-info/RECORD,,