ob-metaflow-extensions 1.4.21__tar.gz → 1.4.30__tar.gz

{ob_metaflow_extensions-1.4.21 → ob_metaflow_extensions-1.4.30}/PKG-INFO

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

{ob_metaflow_extensions-1.4.21 → ob_metaflow_extensions-1.4.30}/metaflow_extensions/outerbounds/plugins/apps/core/app_config.py

@@ -45,10 +45,11 @@ def _try_loading_yaml(file):
 class AuthType:
     BROWSER = "Browser"
     API = "API"
+    BROWSER_AND_API = "BrowserAndApi"
 
     @classmethod
     def enums(cls):
-        return [cls.BROWSER, cls.API]
+        return [cls.BROWSER, cls.API, cls.BROWSER_AND_API]
 
     @classproperty
     def default(cls):

{ob_metaflow_extensions-1.4.21 → ob_metaflow_extensions-1.4.30}/metaflow_extensions/outerbounds/plugins/apps/core/capsule.py

@@ -678,7 +678,7 @@ class CapsuleDeployer:
         auth_type = self._app_config.get_state("auth", {}).get("type", AuthType.default)
         if auth_type == AuthType.BROWSER:
             return "App"
-        elif auth_type == AuthType.API:
+        elif auth_type == AuthType.API or auth_type == AuthType.BROWSER_AND_API:
             return "Endpoint"
         else:
             raise TODOException(f"Unknown auth type: {auth_type}")

{ob_metaflow_extensions-1.4.21 → ob_metaflow_extensions-1.4.30}/metaflow_extensions/outerbounds/plugins/apps/core/code_package/code_packager.py

@@ -381,6 +381,7 @@ class CodePackager:
                     "google-auth",
                     "simple-gcp-object-downloader",
                     "google-cloud-secret-manager",
+                    "packaging",
                 ],
                 "local": [],
             }

{ob_metaflow_extensions-1.4.21 → ob_metaflow_extensions-1.4.30}/metaflow_extensions/outerbounds/plugins/apps/core/config/typed_configs.py

@@ -97,6 +97,7 @@ class TypedCoreConfig:
         branch: Optional[str] = None,
         models: Optional[list] = None,
         data: Optional[list] = None,
+        generate_static_url: Optional[bool] = None,
         **kwargs
     ) -> None:
         self._kwargs = {
@@ -122,6 +123,7 @@ class TypedCoreConfig:
             "branch": branch,
             "models": models,
             "data": data,
+            "generate_static_url": generate_static_url,
         }
         # Add any additional kwargs
         self._kwargs.update(kwargs)

{ob_metaflow_extensions-1.4.21 → ob_metaflow_extensions-1.4.30}/metaflow_extensions/outerbounds/plugins/apps/core/config/unified_config.py

@@ -36,10 +36,11 @@ from .config_utils import (
 class AuthType:
     BROWSER = "Browser"
     API = "API"
+    BROWSER_AND_API = "BrowserAndApi"
 
     @classmethod
     def choices(cls):
-        return [cls.BROWSER, cls.API]
+        return [cls.BROWSER, cls.API, cls.BROWSER_AND_API]
 
 
 class UnitParser:
@@ -966,6 +967,17 @@ How to read this schema:
         is_experimental=True,
         example=[{"asset_id": "data-789", "asset_instance_id": "instance-101"}],
     )
+    generate_static_url = ConfigField(
+        cli_meta=CLIOption(
+            name="generate_static_url",
+            cli_option_str="--generate-static-url",
+            help="Generate a static URL for the app based on its name.",
+            is_flag=True,
+        ),
+        field_type=bool,
+        default=False,
+        help="Generate a static URL for the app based on its name.",
+    )
     # ------- /Experimental -------------
 
     def to_dict(self):

{ob_metaflow_extensions-1.4.21 → ob_metaflow_extensions-1.4.30}/metaflow_extensions/outerbounds/plugins/apps/core/config_schema.yaml

@@ -149,6 +149,7 @@ properties:
         enum:
         - Browser
         - API
+        - BrowserAndApi
         example: Browser
         mutation_behavior: union
         cli_option: --auth-type
@@ -328,3 +329,9 @@ properties:
       asset_instance_id: instance-101
     experimental: true
     mutation_behavior: union
+  generate_static_url:
+    description: Generate a static URL for the app based on its name.
+    type: boolean
+    default: false
+    mutation_behavior: union
+    cli_option: --generate-static-url

{ob_metaflow_extensions-1.4.21 → ob_metaflow_extensions-1.4.30}/metaflow_extensions/outerbounds/plugins/apps/core/experimental/__init__.py

@@ -21,6 +21,10 @@ def capsule_input_overrides(app_config: "AppConfig", capsule_input: dict):
     if persistence is not None and persistence != "none":
         capsule_input["persistence"] = persistence
 
+    capsule_input["generateStaticUrl"] = app_config.get_state(
+        "generate_static_url", False
+    )
+
     model_asset_conf = app_config.get_state("models", None)
     data_asset_conf = app_config.get_state("data", None)
     code_info = _code_info(app_config)

{ob_metaflow_extensions-1.4.21 → ob_metaflow_extensions-1.4.30}/metaflow_extensions/outerbounds/plugins/snowpark/snowpark_client.py

@@ -27,9 +27,12 @@ class SnowparkClient(object):
         except (NameError, ImportError, ModuleNotFoundError):
             raise SnowflakeException(
                 "Could not import module 'snowflake'.\n\nInstall Snowflake "
-                "Python package (https://pypi.org/project/snowflake/) first.\n"
-                "You can install the module by executing - "
-                "%s -m pip install snowflake\n"
+                "Python packages first:\n"
+                "  snowflake==1.8.0\n"
+                "  snowflake-connector-python==3.18.0\n"
+                "  snowflake-snowpark-python==1.40.0\n\n"
+                "You can install them by executing:\n"
+                "%s -m pip install snowflake==1.8.0 snowflake-connector-python==3.18.0 snowflake-snowpark-python==1.40.0\n"
                 "or equivalent through your favorite Python package manager."
                 % sys.executable
             )

{ob_metaflow_extensions-1.4.21 → ob_metaflow_extensions-1.4.30}/metaflow_extensions/outerbounds/plugins/snowpark/snowpark_decorator.py

@@ -42,10 +42,13 @@ class Snowflake(object):
             return session
         except (NameError, ImportError, ModuleNotFoundError):
             raise SnowflakeException(
-                "Could not import module 'snowflake'.\n\nInstall Snowflake "
-                "Python package (https://pypi.org/project/snowflake/) first.\n"
-                "You can install the module by using the @pypi decorator - "
-                "Eg: @pypi(packages={'snowflake': '0.11.0'})\n"
+                "Could not import module 'snowflake'.\n\n"
+                "Install required Snowflake packages using the @pypi decorator:\n"
+                "@pypi(packages={\n"
+                "    'snowflake': '1.8.0',\n"
+                "    'snowflake-connector-python': '3.18.0',\n"
+                "    'snowflake-snowpark-python': '1.40.0'\n"
+                "})\n"
             )
 
 
@@ -143,9 +146,12 @@ class SnowparkDecorator(StepDecorator):
         except (NameError, ImportError, ModuleNotFoundError):
             raise SnowflakeException(
                 "Could not import module 'snowflake'.\n\nInstall Snowflake "
-                "Python package (https://pypi.org/project/snowflake/) first.\n"
-                "You can install the module by executing - "
-                "%s -m pip install snowflake\n"
+                "Python packages first:\n"
+                "  snowflake==1.8.0\n"
+                "  snowflake-connector-python==3.18.0\n"
+                "  snowflake-snowpark-python==1.40.0\n\n"
+                "You can install them by executing:\n"
+                "%s -m pip install snowflake==1.8.0 snowflake-connector-python==3.18.0 snowflake-snowpark-python==1.40.0\n"
                 "or equivalent through your favorite Python package manager."
                 % sys.executable
             )

{ob_metaflow_extensions-1.4.21 → ob_metaflow_extensions-1.4.30}/metaflow_extensions/outerbounds/plugins/snowpark/snowpark_job.py

@@ -199,11 +199,17 @@ class RunningJob(object):
 
     @property
     def status(self):
-        return self.status_obj()[0].get("status")
+        status_list = self.status_obj()
+        if not status_list:
+            return "UNKNOWN"
+        return status_list[0].get("status", "UNKNOWN")
 
     @property
     def message(self):
-        return self.status_obj()[0].get("message")
+        status_list = self.status_obj()
+        if not status_list:
+            return None
+        return status_list[0].get("message")
 
     @property
     def is_waiting(self):

{ob_metaflow_extensions-1.4.21 → ob_metaflow_extensions-1.4.30}/ob_metaflow_extensions.egg-info/PKG-INFO

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

ob_metaflow_extensions-1.4.30/ob_metaflow_extensions.egg-info/requires.txt

@@ -0,0 +1,3 @@
+boto3
+kubernetes
+ob-metaflow==2.19.5.1

{ob_metaflow_extensions-1.4.21 → ob_metaflow_extensions-1.4.30}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_namespace_packages
 from pathlib import Path
 
 
-version = "1.4.21"
+version = "1.4.30"
 this_directory = Path(__file__).parent
 long_description = (this_directory / "README.md").read_text()
 
@@ -19,6 +19,6 @@ setup(
     install_requires=[
         "boto3",
         "kubernetes",
-        "ob-metaflow == 2.18.10.1",
+        "ob-metaflow == 2.19.5.1",
     ],
 )

ob_metaflow_extensions-1.4.21/ob_metaflow_extensions.egg-info/requires.txt

@@ -1,3 +0,0 @@
-boto3
-kubernetes
-ob-metaflow==2.18.10.1