ob-metaflow-extensions 1.2.5rc0__tar.gz → 1.2.7rc0__tar.gz

{ob_metaflow_extensions-1.2.5rc0 → ob_metaflow_extensions-1.2.7rc0}/PKG-INFO

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

{ob_metaflow_extensions-1.2.5rc0 → ob_metaflow_extensions-1.2.7rc0}/metaflow_extensions/outerbounds/plugins/aws/assume_role_decorator.py

@@ -1,10 +1,12 @@
-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 metaflow.user_configs.config_decorators import (
+    MutableFlow,
+    MutableStep,
+    CustomFlowDecorator,
+)
 from .assume_role import OBP_ASSUME_ROLE_ARN_ENV_VAR
 
 
-class assume_role(FlowMutator):
+class assume_role(CustomFlowDecorator):
     """
     Flow-level decorator for assuming AWS IAM roles.
 
@@ -40,7 +42,7 @@ class assume_role(FlowMutator):
                 "`role_arn` must be a valid AWS IAM role ARN starting with 'arn:aws:iam::'"
             )
 
-    def pre_mutate(self, mutable_flow: MutableFlow) -> None:
+    def evaluate(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.5rc0 → ob_metaflow_extensions-1.2.7rc0}/metaflow_extensions/outerbounds/plugins/checkpoint_datastores/coreweave.py

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

{ob_metaflow_extensions-1.2.5rc0 → ob_metaflow_extensions-1.2.7rc0}/metaflow_extensions/outerbounds/plugins/checkpoint_datastores/nebius.py

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

{ob_metaflow_extensions-1.2.5rc0 → ob_metaflow_extensions-1.2.7rc0}/metaflow_extensions/outerbounds/plugins/fast_bakery/docker_environment.py

@@ -351,16 +351,12 @@ class DockerEnvironment(MetaflowEnvironment):
         config.append("--disable=F0401")
         return config
 
-    def get_package_commands(
-        self, codepackage_url, datastore_type, code_package_metadata=None
-    ):
+    def get_package_commands(self, codepackage_url, datastore_type):
         # 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, code_package_metadata=code_package_metadata
-        )
+        ] + super().get_package_commands(codepackage_url, datastore_type)
 
     def bootstrap_commands(self, step_name, datastore_type):
         if step_name in self.skipped_steps:

{ob_metaflow_extensions-1.2.5rc0 → ob_metaflow_extensions-1.2.7rc0}/ob_metaflow_extensions.egg-info/PKG-INFO

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

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

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

{ob_metaflow_extensions-1.2.5rc0 → ob_metaflow_extensions-1.2.7rc0}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_namespace_packages
 from pathlib import Path
 
 
-version = "1.2.5rc0"
+version = "1.2.7rc0"
 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.16.1.1",
+        "ob-metaflow == 2.15.21.2",
     ],
 )

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

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