ob-metaflow-extensions 1.2.0__tar.gz → 1.2.2rc0__tar.gz

{ob_metaflow_extensions-1.2.0 → ob_metaflow_extensions-1.2.2rc0}/PKG-INFO

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

{ob_metaflow_extensions-1.2.0 → ob_metaflow_extensions-1.2.2rc0}/metaflow_extensions/outerbounds/plugins/aws/assume_role_decorator.py

@@ -1,12 +1,10 @@
-from metaflow.user_configs.config_decorators import (
-    MutableFlow,
-    MutableStep,
-    CustomFlowDecorator,
-)
+from metaflow.user_decorators.user_flow_decorator import FlowMutator
+from metaflow.user_decorators.mutable_flow import MutableFlow
+from metaflow.user_decorators.mutable_step import MutableStep
 from .assume_role import OBP_ASSUME_ROLE_ARN_ENV_VAR
 
 
-class assume_role(CustomFlowDecorator):
+class assume_role(FlowMutator):
     """
     Flow-level decorator for assuming AWS IAM roles.
 
@@ -42,7 +40,7 @@ class assume_role(CustomFlowDecorator):
                 "`role_arn` must be a valid AWS IAM role ARN starting with 'arn:aws:iam::'"
             )
 
-    def evaluate(self, mutable_flow: MutableFlow) -> None:
+    def pre_mutate(self, mutable_flow: MutableFlow) -> None:
         """
         This method is called by Metaflow to apply the decorator to the flow.
         It sets up environment variables that will be used by the AWS client

{ob_metaflow_extensions-1.2.0 → ob_metaflow_extensions-1.2.2rc0}/metaflow_extensions/outerbounds/plugins/checkpoint_datastores/coreweave.py

@@ -1,12 +1,10 @@
-from metaflow.user_configs.config_decorators import (
-    MutableFlow,
-    MutableStep,
-    CustomFlowDecorator,
-)
+from metaflow.user_decorators.user_flow_decorator import FlowMutator
+from metaflow.user_decorators.mutable_flow import MutableFlow
+from metaflow.user_decorators.mutable_step import MutableStep
 import os
 
 
-class coreweave_checkpoints(CustomFlowDecorator):
+class coreweave_checkpoints(FlowMutator):
 
     """
 
@@ -28,7 +26,7 @@ class coreweave_checkpoints(CustomFlowDecorator):
     ```python
     from metaflow import checkpoint, step, FlowSpec, coreweave_checkpoints
 
-    @coreweave_checkpoints(secrets=[], bucket_path=None)
+    @coreweave_checkpoints(secrets=[], bucket_path="s3://my-coreweave-bucket/foo")
     class MyFlow(FlowSpec):
         @checkpoint
         @step
@@ -42,9 +40,6 @@ class coreweave_checkpoints(CustomFlowDecorator):
     ```
     """
 
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-
     def init(self, *args, **kwargs):
         self.bucket_path = kwargs.get("bucket_path", None)
 
@@ -64,7 +59,7 @@ class coreweave_checkpoints(CustomFlowDecorator):
                 "`secrets` keyword argument is required for the coreweave_datastore"
             )
 
-    def evaluate(self, mutable_flow: MutableFlow) -> None:
+    def pre_mutate(self, mutable_flow: MutableFlow) -> None:
         from metaflow import (
             checkpoint,
             model,

{ob_metaflow_extensions-1.2.0 → ob_metaflow_extensions-1.2.2rc0}/metaflow_extensions/outerbounds/plugins/checkpoint_datastores/nebius.py

@@ -1,14 +1,12 @@
-from metaflow.user_configs.config_decorators import (
-    MutableFlow,
-    MutableStep,
-    CustomFlowDecorator,
-)
+from metaflow.user_decorators.user_flow_decorator import FlowMutator
+from metaflow.user_decorators.mutable_flow import MutableFlow
+from metaflow.user_decorators.mutable_step import MutableStep
 import os
 
 NEBIUS_ENDPOINT_URL = "https://storage.eu-north1.nebius.cloud:443"
 
 
-class nebius_checkpoints(CustomFlowDecorator):
+class nebius_checkpoints(FlowMutator):
 
     """
 
@@ -34,7 +32,7 @@ class nebius_checkpoints(CustomFlowDecorator):
     ```python
     from metaflow import checkpoint, step, FlowSpec, nebius_checkpoints
 
-    @nebius_checkpoints(secrets=[], bucket_path=None)
+    @nebius_checkpoints(secrets=[], bucket_path="s3://my-nebius-bucket/foo")
     class MyFlow(FlowSpec):
         @checkpoint
         @step
@@ -48,9 +46,6 @@ class nebius_checkpoints(CustomFlowDecorator):
     ```
     """
 
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-
     def init(self, *args, **kwargs):
         self.bucket_path = kwargs.get("bucket_path", None)
 
@@ -70,7 +65,7 @@ class nebius_checkpoints(CustomFlowDecorator):
                 "`secrets` keyword argument is required for the coreweave_datastore"
             )
 
-    def evaluate(self, mutable_flow: MutableFlow) -> None:
+    def pre_mutate(self, mutable_flow: MutableFlow) -> None:
         from metaflow import (
             checkpoint,
             model,

{ob_metaflow_extensions-1.2.0 → ob_metaflow_extensions-1.2.2rc0}/metaflow_extensions/outerbounds/plugins/fast_bakery/docker_environment.py

@@ -351,12 +351,16 @@ class DockerEnvironment(MetaflowEnvironment):
         config.append("--disable=F0401")
         return config
 
-    def get_package_commands(self, codepackage_url, datastore_type):
+    def get_package_commands(
+        self, codepackage_url, datastore_type, code_package_metadata=None
+    ):
         # we must set the skip install flag at this stage in order to skip package downloads,
         # doing so in bootstrap_commands is too late in the lifecycle.
         return [
             "export METAFLOW_SKIP_INSTALL_DEPENDENCIES=$FASTBAKERY_IMAGE",
-        ] + super().get_package_commands(codepackage_url, datastore_type)
+        ] + super().get_package_commands(
+            codepackage_url, datastore_type, code_package_metadata=code_package_metadata
+        )
 
     def bootstrap_commands(self, step_name, datastore_type):
         if step_name in self.skipped_steps:

{ob_metaflow_extensions-1.2.0 → ob_metaflow_extensions-1.2.2rc0}/ob_metaflow_extensions.egg-info/PKG-INFO

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

ob_metaflow_extensions-1.2.2rc0/ob_metaflow_extensions.egg-info/requires.txt

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

{ob_metaflow_extensions-1.2.0 → ob_metaflow_extensions-1.2.2rc0}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_namespace_packages
 from pathlib import Path
 
 
-version = "1.2.0"
+version = "1.2.2rc0"
 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.15.21.1",
+        "ob-metaflow == 2.16.0.1",
     ],
 )

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

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