ob-metaflow-extensions 1.2.7rc1__py2.py3-none-any.whl → 1.2.9__py2.py3-none-any.whl

metaflow_extensions/outerbounds/plugins/__init__.py

@@ -335,6 +335,7 @@ STEP_DECORATORS_DESC = [
     ("snowpark", ".snowpark.snowpark_decorator.SnowparkDecorator"),
     ("tensorboard", ".tensorboard.TensorboardDecorator"),
     ("gpu_profile", ".profilers.gpu_profile_decorator.GPUProfileDecorator"),
+    ("test_append_card", ".profilers.simple_card_decorator.DynamicCardAppendDecorator"),
     ("nim", ".nim.nim_decorator.NimDecorator"),
     ("ollama", ".ollama.OllamaDecorator"),
     ("vllm", ".vllm.VLLMDecorator"),

metaflow_extensions/outerbounds/plugins/profilers/simple_card_decorator.py

@@ -0,0 +1,96 @@
+from datetime import datetime
+from metaflow.decorators import StepDecorator
+from ..card_utilities.injector import CardDecoratorInjector
+
+
+class DynamicCardAppendDecorator(StepDecorator):
+    """
+    A simple decorator that demonstrates using CardDecoratorInjector
+    to inject a card and render simple markdown content.
+    """
+
+    name = "test_append_card"
+
+    defaults = {
+        "title": "Simple Card",
+        "message": "Hello from DynamicCardAppendDecorator!",
+        "show_timestamp": True,
+        "refresh_interval": 5,
+    }
+
+    CARD_ID = "simple_card"
+
+    def step_init(
+        self, flow, graph, step_name, decorators, environment, flow_datastore, logger
+    ):
+        """Initialize the decorator and inject the card."""
+        self.deco_injector = CardDecoratorInjector()
+        self.deco_injector.attach_card_decorator(
+            flow,
+            step_name,
+            self.CARD_ID,
+            "blank",
+            refresh_interval=self.attributes["refresh_interval"],
+        )
+
+    def task_decorate(
+        self, step_func, flow, graph, retry_count, max_user_code_retries, ubf_context
+    ):
+        """Decorate the step function to add card content."""
+        from metaflow import current
+        from metaflow.cards import Markdown
+
+        # Create the card content
+        title = self.attributes["title"]
+        message = self.attributes["message"]
+        show_timestamp = self.attributes["show_timestamp"]
+
+        # Add title to the card
+        current.card[self.CARD_ID].append(Markdown(f"# {title}"))
+
+        # Add message to the card
+        current.card[self.CARD_ID].append(Markdown(f"**Message:** {message}"))
+
+        # Add timestamp if requested
+        if show_timestamp:
+            timestamp = datetime.now().astimezone().strftime("%Y-%m-%d %H:%M:%S %z")
+            current.card[self.CARD_ID].append(Markdown(f"**Created at:** {timestamp}"))
+
+        # Add step information
+        current.card[self.CARD_ID].append(Markdown(f"**Step:** `{current.pathspec}`"))
+
+        # Add a simple divider
+        current.card[self.CARD_ID].append(Markdown("---"))
+
+        # Add some dynamic content that shows this is working
+        current.card[self.CARD_ID].append(
+            Markdown("**Status:** Card successfully injected! 🎉")
+        )
+
+        def wrapped_step_func():
+            """Execute the original step function."""
+            try:
+                # Before execution
+                current.card[self.CARD_ID].append(
+                    Markdown("**Execution:** Step started...")
+                )
+                current.card[self.CARD_ID].refresh()
+
+                # Execute the original step
+                step_func()
+
+                # After execution
+                current.card[self.CARD_ID].append(
+                    Markdown("**Execution:** Step completed successfully! ✅")
+                )
+                current.card[self.CARD_ID].refresh()
+
+            except Exception as e:
+                # Handle errors
+                current.card[self.CARD_ID].append(
+                    Markdown(f"**Error:** Step failed with error: `{str(e)}` ❌")
+                )
+                current.card[self.CARD_ID].refresh()
+                raise
+
+        return wrapped_step_func

{ob_metaflow_extensions-1.2.7rc1.dist-info → ob_metaflow_extensions-1.2.9.dist-info}/METADATA

@@ -1,13 +1,13 @@
 Metadata-Version: 2.1
 Name: ob-metaflow-extensions
-Version: 1.2.7rc1
+Version: 1.2.9
 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.15.21.2)
+Requires-Dist: ob-metaflow (==2.15.21.5)
 
 # Outerbounds platform package
 

{ob_metaflow_extensions-1.2.7rc1.dist-info → ob_metaflow_extensions-1.2.9.dist-info}/RECORD

@@ -1,7 +1,7 @@
 metaflow_extensions/outerbounds/__init__.py,sha256=Gb8u06s9ClQsA_vzxmkCzuMnigPy7kKcDnLfb7eB-64,514
 metaflow_extensions/outerbounds/remote_config.py,sha256=pEFJuKDYs98eoB_-ryPjVi9b_c4gpHMdBHE14ltoxIU,4672
 metaflow_extensions/outerbounds/config/__init__.py,sha256=JsQGRuGFz28fQWjUvxUgR8EKBLGRdLUIk_buPLJplJY,1225
-metaflow_extensions/outerbounds/plugins/__init__.py,sha256=SgqRTG7FFk-jrPBNPYQ-D0W26Yr_dgg7ZL-YJInTQGw,13706
+metaflow_extensions/outerbounds/plugins/__init__.py,sha256=c4zZnULrBOU2bmleU_D7Xr5H3kTuTgDhtZIKUAJiNqw,13795
 metaflow_extensions/outerbounds/plugins/auth_server.py,sha256=_Q9_2EL0Xy77bCRphkwT1aSu8gQXRDOH-Z-RxTUO8N4,2202
 metaflow_extensions/outerbounds/plugins/perimeters.py,sha256=QXh3SFP7GQbS-RAIxUOPbhPzQ7KDFVxZkTdKqFKgXjI,2697
 metaflow_extensions/outerbounds/plugins/apps/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -84,6 +84,7 @@ metaflow_extensions/outerbounds/plugins/ollama/ollama.py,sha256=C-6Hz8OxsJiB14AA
 metaflow_extensions/outerbounds/plugins/ollama/status_card.py,sha256=F5e4McDl28lhtjeUyInkl03bqjr1lgLxWoau8Q9xwBE,10994
 metaflow_extensions/outerbounds/plugins/profilers/deco_injector.py,sha256=oI_C3c64XBm7n88FILqHwn-Nnc5DeT_68I67lM9rXaI,2434
 metaflow_extensions/outerbounds/plugins/profilers/gpu_profile_decorator.py,sha256=gDHQ2sMIp4NuZSzUspbSd8RGdFAoO5mgZAyFcZ2a51Y,2619
+metaflow_extensions/outerbounds/plugins/profilers/simple_card_decorator.py,sha256=4W9tLGCmkFx-4XYLa1xF6qMiaWOBYYFx_RclZDKej30,3259
 metaflow_extensions/outerbounds/plugins/secrets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 metaflow_extensions/outerbounds/plugins/secrets/secrets.py,sha256=3s98hO_twKkM22tKyDdcUjGQNfYpSXW_jLKISV9ju_U,8433
 metaflow_extensions/outerbounds/plugins/snowflake/__init__.py,sha256=RG4ixt3jwqcK1_tt0QxLcUbNmf7wWAMnZhBx-ZMGgLk,114
@@ -115,7 +116,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.7rc1.dist-info/METADATA,sha256=uxDptIcQbq__3AinjXnU6P2BfNqpiFAp0yBMQoR8m6k,522
-ob_metaflow_extensions-1.2.7rc1.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
-ob_metaflow_extensions-1.2.7rc1.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
-ob_metaflow_extensions-1.2.7rc1.dist-info/RECORD,,
+ob_metaflow_extensions-1.2.9.dist-info/METADATA,sha256=-i0hbI1ZXbWZmHbP67Kq7ibjnj_sh_m1cEGwvU0ew2w,519
+ob_metaflow_extensions-1.2.9.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
+ob_metaflow_extensions-1.2.9.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
+ob_metaflow_extensions-1.2.9.dist-info/RECORD,,