ob-metaflow-extensions 1.2.5rc0__py2.py3-none-any.whl → 1.2.7__py2.py3-none-any.whl

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

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,

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,

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.dist-info → ob_metaflow_extensions-1.2.7.dist-info}/METADATA

@@ -1,13 +1,13 @@
 Metadata-Version: 2.1
 Name: ob-metaflow-extensions
-Version: 1.2.5rc0
+Version: 1.2.7
 Summary: Outerbounds Platform Extensions for Metaflow
 Author: Outerbounds, Inc.
 License: Commercial
 Description-Content-Type: text/markdown
 Requires-Dist: boto3
 Requires-Dist: kubernetes
-Requires-Dist: ob-metaflow (==2.16.1.1)
+Requires-Dist: ob-metaflow (==2.15.21.3)
 
 # Outerbounds platform package
 

{ob_metaflow_extensions-1.2.5rc0.dist-info → ob_metaflow_extensions-1.2.7.dist-info}/RECORD

@@ -41,17 +41,17 @@ metaflow_extensions/outerbounds/plugins/apps/core/config/unified_config.py,sha25
 metaflow_extensions/outerbounds/plugins/apps/core/experimental/__init__.py,sha256=rd4qGTkHndKYfJmoAKZWiY0KK4j5BK6RBrtle-it1Mg,2746
 metaflow_extensions/outerbounds/plugins/aws/__init__.py,sha256=VBGdjNKeFLXGZuqh4jVk8cFtO1AWof73a6k_cnbAOYA,145
 metaflow_extensions/outerbounds/plugins/aws/assume_role.py,sha256=mBewNlnSYsR2rFXFkX-DUH6ku01h2yOcMcLHoCL7eyI,161
-metaflow_extensions/outerbounds/plugins/aws/assume_role_decorator.py,sha256=nkolUdV09ieFGrAQWgQ1lsoiIKpkIrfYvh4lEyQBXSs,2463
+metaflow_extensions/outerbounds/plugins/aws/assume_role_decorator.py,sha256=b3te8-RJxsj5cqXbQTFxxeIfnqTERAFInzLgXLVLyYA,2391
 metaflow_extensions/outerbounds/plugins/card_utilities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 metaflow_extensions/outerbounds/plugins/card_utilities/async_cards.py,sha256=6rQhtZXK5DenXPfCRS1ul0jvLJYd48jrJAlnodID21w,4347
 metaflow_extensions/outerbounds/plugins/card_utilities/extra_components.py,sha256=D8rgCaSc7PuLD0MHJjqsjN0g0PQMN1H-ySOJqi5uIOE,19111
 metaflow_extensions/outerbounds/plugins/card_utilities/injector.py,sha256=UQtHrviFs7Z5LBh0nbaIv8aEnHnyXFjs9TiGCjtcIWc,2452
 metaflow_extensions/outerbounds/plugins/checkpoint_datastores/__init__.py,sha256=H96Bdv2XKTewKIU587sjNbU538SKMZOYWQ2qCp1JaNU,84
-metaflow_extensions/outerbounds/plugins/checkpoint_datastores/coreweave.py,sha256=WVpm1baDIen61IdHijVqHJ02NlRzu1tCw3cuOfqmoIw,4548
-metaflow_extensions/outerbounds/plugins/checkpoint_datastores/nebius.py,sha256=fVhIOYxP0A_rHhKhybMScEp5XjkQdmMUws_u1m4G25w,4670
+metaflow_extensions/outerbounds/plugins/checkpoint_datastores/coreweave.py,sha256=_WzoOROFjoFa8TzsMNFp-r_1Zz7NUp-5ljn_kKlczXA,4534
+metaflow_extensions/outerbounds/plugins/checkpoint_datastores/nebius.py,sha256=zgqDLFewCeF5jqh-hUNKmC_OAjld09ln0bb8Lkeqapc,4659
 metaflow_extensions/outerbounds/plugins/fast_bakery/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 metaflow_extensions/outerbounds/plugins/fast_bakery/baker.py,sha256=ShE5omFBr83wkvEhL_ptRFvDNMs6wefg4BjaafQjTcM,3602
-metaflow_extensions/outerbounds/plugins/fast_bakery/docker_environment.py,sha256=vKN3Waz0SgbOALVTRcf_8MCGhxdRueRFZJ7JmWbRJRM,15487
+metaflow_extensions/outerbounds/plugins/fast_bakery/docker_environment.py,sha256=nmp_INGIAiWyrhyJ71BH38eRLu1xCIEEKejmXNQ6RlA,15378
 metaflow_extensions/outerbounds/plugins/fast_bakery/fast_bakery.py,sha256=PE81ZB54OAMXkMGSB7JqgvgMg7N9kvoVclrWL-6jc2U,5626
 metaflow_extensions/outerbounds/plugins/fast_bakery/fast_bakery_cli.py,sha256=kqFyu2bJSnc9_9aYfBpz5xK6L6luWFZK_NMuh8f1eVk,1494
 metaflow_extensions/outerbounds/plugins/fast_bakery/fast_bakery_decorator.py,sha256=MXSIp05-jvt8Q2uGaLKjtuM_ToLeRLxhtMbfHc9Kcko,1515
@@ -115,7 +115,7 @@ metaflow_extensions/outerbounds/toplevel/plugins/ollama/__init__.py,sha256=GRSz2
 metaflow_extensions/outerbounds/toplevel/plugins/snowflake/__init__.py,sha256=LptpH-ziXHrednMYUjIaosS1SXD3sOtF_9_eRqd8SJw,50
 metaflow_extensions/outerbounds/toplevel/plugins/torchtune/__init__.py,sha256=uTVkdSk3xZ7hEKYfdlyVteWj5KeDwaM1hU9WT-_YKfI,50
 metaflow_extensions/outerbounds/toplevel/plugins/vllm/__init__.py,sha256=ekcgD3KVydf-a0xMI60P4uy6ePkSEoFHiGnDq1JM940,45
-ob_metaflow_extensions-1.2.5rc0.dist-info/METADATA,sha256=0sXyM3KJUplG2UCjyy__LMM6xjZjfgVBwIfCbmWXs3k,521
-ob_metaflow_extensions-1.2.5rc0.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
-ob_metaflow_extensions-1.2.5rc0.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
-ob_metaflow_extensions-1.2.5rc0.dist-info/RECORD,,
+ob_metaflow_extensions-1.2.7.dist-info/METADATA,sha256=FwXwSPHIJx2B3vtETT7lLhl1iyCzwtQLv2BEc-rNsB4,519
+ob_metaflow_extensions-1.2.7.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
+ob_metaflow_extensions-1.2.7.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
+ob_metaflow_extensions-1.2.7.dist-info/RECORD,,