ob-metaflow-extensions 1.1.171rc1__tar.gz → 1.1.172__tar.gz

{ob_metaflow_extensions-1.1.171rc1 → ob_metaflow_extensions-1.1.172}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob_metaflow_extensions
-Version: 1.1.171rc1
+Version: 1.1.172
 Summary: Outerbounds Platform Extensions for Metaflow
 Author: Outerbounds, Inc.
 License: Commercial

{ob_metaflow_extensions-1.1.171rc1 → ob_metaflow_extensions-1.1.172}/metaflow_extensions/outerbounds/plugins/__init__.py

@@ -324,7 +324,6 @@ CLIS_DESC = [
     ("nvct", ".nvct.nvct_cli.cli"),
     ("fast-bakery", ".fast_bakery.fast_bakery_cli.cli"),
     ("snowpark", ".snowpark.snowpark_cli.cli"),
-    ("app", ".apps.app_cli.cli"),
 ]
 STEP_DECORATORS_DESC = [
     ("nvidia", ".nvcf.nvcf_decorator.NvcfDecorator"),

{ob_metaflow_extensions-1.1.171rc1 → ob_metaflow_extensions-1.1.172}/metaflow_extensions/outerbounds/plugins/nvct/nvct_decorator.py

@@ -46,23 +46,23 @@ SUPPORTABLE_GPU_TYPES = {
     "H100": [
         {
             "n_gpus": 1,
-            "instance_type": "GCP.GPU.H100_1x",
-            "backend": "gcp-asia-se-1a",
+            "instance_type": "OCI.GPU.H100_1x",
+            "backend": "nvcf-dgxc-k8s-oci-nrt-prd8",
         },
         {
             "n_gpus": 2,
-            "instance_type": "GCP.GPU.H100_2x",
-            "backend": "gcp-asia-se-1a",
+            "instance_type": "OCI.GPU.H100_2x",
+            "backend": "nvcf-dgxc-k8s-oci-nrt-prd8",
         },
         {
             "n_gpus": 4,
-            "instance_type": "GCP.GPU.H100_4x",
-            "backend": "gcp-asia-se-1a",
+            "instance_type": "OCI.GPU.H100_4x",
+            "backend": "nvcf-dgxc-k8s-oci-nrt-prd8",
         },
         {
             "n_gpus": 8,
-            "instance_type": "GCP.GPU.H100_8x",
-            "backend": "gcp-asia-se-1a",
+            "instance_type": "OCI.GPU.H100_8x",
+            "backend": "nvcf-dgxc-k8s-oci-nrt-prd8",
         },
     ],
     "NEBIUS_H100": [

{ob_metaflow_extensions-1.1.171rc1 → ob_metaflow_extensions-1.1.172}/metaflow_extensions/outerbounds/toplevel/ob_internal.py

@@ -1,2 +1 @@
 from ..plugins.kubernetes.pod_killer import PodKiller
-from ..plugins.fast_bakery.baker import bake_image

{ob_metaflow_extensions-1.1.171rc1 → ob_metaflow_extensions-1.1.172}/ob_metaflow_extensions.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob-metaflow-extensions
-Version: 1.1.171rc1
+Version: 1.1.172
 Summary: Outerbounds Platform Extensions for Metaflow
 Author: Outerbounds, Inc.
 License: Commercial

{ob_metaflow_extensions-1.1.171rc1 → ob_metaflow_extensions-1.1.172}/ob_metaflow_extensions.egg-info/SOURCES.txt

@@ -7,7 +7,6 @@ metaflow_extensions/outerbounds/plugins/__init__.py
 metaflow_extensions/outerbounds/plugins/auth_server.py
 metaflow_extensions/outerbounds/plugins/perimeters.py
 metaflow_extensions/outerbounds/plugins/apps/__init__.py
-metaflow_extensions/outerbounds/plugins/apps/app_cli.py
 metaflow_extensions/outerbounds/plugins/apps/app_utils.py
 metaflow_extensions/outerbounds/plugins/apps/consts.py
 metaflow_extensions/outerbounds/plugins/apps/deploy_decorator.py
@@ -23,7 +22,6 @@ metaflow_extensions/outerbounds/plugins/checkpoint_datastores/__init__.py
 metaflow_extensions/outerbounds/plugins/checkpoint_datastores/coreweave.py
 metaflow_extensions/outerbounds/plugins/checkpoint_datastores/nebius.py
 metaflow_extensions/outerbounds/plugins/fast_bakery/__init__.py
-metaflow_extensions/outerbounds/plugins/fast_bakery/baker.py
 metaflow_extensions/outerbounds/plugins/fast_bakery/docker_environment.py
 metaflow_extensions/outerbounds/plugins/fast_bakery/fast_bakery.py
 metaflow_extensions/outerbounds/plugins/fast_bakery/fast_bakery_cli.py

{ob_metaflow_extensions-1.1.171rc1 → ob_metaflow_extensions-1.1.172}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_namespace_packages
 from pathlib import Path
 
 
-version = "1.1.171rc1"
+version = "1.1.172"
 this_directory = Path(__file__).parent
 long_description = (this_directory / "README.md").read_text()
 

ob_metaflow_extensions-1.1.171rc1/metaflow_extensions/outerbounds/plugins/apps/app_cli.py

@@ -1,29 +0,0 @@
-from metaflow._vendor import click
-import os
-
-os.environ["APPS_CLI_LOADING_IN_METAFLOW"] = "true"
-OUTERBOUNDS_APP_CLI_AVAILABLE = True
-
-try:
-    import outerbounds.apps.app_cli as ob_apps_cli
-except ImportError:
-    OUTERBOUNDS_APP_CLI_AVAILABLE = False
-
-
-if not OUTERBOUNDS_APP_CLI_AVAILABLE:
-
-    @click.group()
-    def _cli():
-        pass
-
-    @_cli.group(help="Dummy Group to append to CLI for Safety")
-    def app():
-        pass
-
-    @app.command(help="Dummy Command to append to CLI for Safety")
-    def cannot_deploy():
-        raise Exception("Outerbounds App CLI not available")
-
-    cli = _cli
-else:
-    cli = ob_apps_cli.cli

ob_metaflow_extensions-1.1.171rc1/metaflow_extensions/outerbounds/plugins/fast_bakery/baker.py

@@ -1,110 +0,0 @@
-import threading
-import time
-import sys
-from typing import Dict, Optional, Any, Callable
-from functools import partial
-from metaflow.exception import MetaflowException
-from metaflow.metaflow_config import FAST_BAKERY_URL
-
-from .fast_bakery import FastBakery, FastBakeryApiResponse, FastBakeryException
-from .docker_environment import cache_request
-
-BAKERY_METAFILE = ".imagebakery-cache"
-
-
-class BakerException(MetaflowException):
-    headline = "Ran into an error while baking image"
-
-    def __init__(self, msg):
-        super(BakerException, self).__init__(msg)
-
-
-def bake_image(
-    cache_file_path: str,
-    ref: Optional[str] = None,
-    python: Optional[str] = None,
-    pypi_packages: Optional[Dict[str, str]] = None,
-    conda_packages: Optional[Dict[str, str]] = None,
-    base_image: Optional[str] = None,
-    logger: Optional[Callable[[str], Any]] = None,
-) -> FastBakeryApiResponse:
-    """
-    Bakes a Docker image with the specified dependencies.
-
-    Args:
-        cache_file_path: Path to the cache file
-        ref: Reference identifier for this bake (for logging purposes)
-        python: Python version to use
-        pypi_packages: Dictionary of PyPI packages and versions
-        conda_packages: Dictionary of Conda packages and versions
-        base_image: Base Docker image to use
-        logger: Optional logger function to output progress
-
-    Returns:
-        FastBakeryApiResponse: The response from the bakery service
-
-    Raises:
-        BakerException: If the baking process fails
-    """
-    # Default logger if none provided
-    if logger is None:
-        logger = partial(print, file=sys.stderr)
-
-    # Thread lock for logging
-    logger_lock = threading.Lock()
-    images_baked = 0
-
-    @cache_request(cache_file_path)
-    def _cached_bake(
-        ref=None,
-        python=None,
-        pypi_packages=None,
-        conda_packages=None,
-        base_image=None,
-    ):
-        try:
-            bakery = FastBakery(url=FAST_BAKERY_URL)
-            bakery._reset_payload()
-            bakery.python_version(python)
-            bakery.pypi_packages(pypi_packages)
-            bakery.conda_packages(conda_packages)
-            bakery.base_image(base_image)
-            # bakery.ignore_cache()
-
-            with logger_lock:
-                logger(f"🍳 Baking [{ref}] ...")
-                logger(f"     🐍 Python: {python}")
-
-                if pypi_packages:
-                    logger(f"     📦 PyPI packages:")
-                    for package, version in pypi_packages.items():
-                        logger(f"        🔧 {package}: {version}")
-
-                if conda_packages:
-                    logger(f"     📦 Conda packages:")
-                    for package, version in conda_packages.items():
-                        logger(f"        🔧 {package}: {version}")
-
-                logger(f"     🏗️  Base image: {base_image}")
-
-            start_time = time.time()
-            res = bakery.bake()
-            # TODO: Get actual bake time from bakery
-            bake_time = time.time() - start_time
-
-            with logger_lock:
-                logger(f"🏁 Baked [{ref}] in {bake_time:.2f} seconds!")
-            nonlocal images_baked
-            images_baked += 1
-            return res
-        except FastBakeryException as ex:
-            raise BakerException(f"Bake [{ref}] failed: {str(ex)}")
-
-    # Call the cached bake function with the provided parameters
-    return _cached_bake(
-        ref=ref,
-        python=python,
-        pypi_packages=pypi_packages,
-        conda_packages=conda_packages,
-        base_image=base_image,
-    )