ob-metaflow-extensions 1.2.7rc1__tar.gz → 1.2.8__tar.gz

{ob_metaflow_extensions-1.2.7rc1 → ob_metaflow_extensions-1.2.8}/PKG-INFO

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

{ob_metaflow_extensions-1.2.7rc1 → ob_metaflow_extensions-1.2.8}/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"),

ob_metaflow_extensions-1.2.8/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 → ob_metaflow_extensions-1.2.8}/ob_metaflow_extensions.egg-info/PKG-INFO

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

{ob_metaflow_extensions-1.2.7rc1 → ob_metaflow_extensions-1.2.8}/ob_metaflow_extensions.egg-info/SOURCES.txt

@@ -87,6 +87,7 @@ metaflow_extensions/outerbounds/plugins/ollama/ollama.py
 metaflow_extensions/outerbounds/plugins/ollama/status_card.py
 metaflow_extensions/outerbounds/plugins/profilers/deco_injector.py
 metaflow_extensions/outerbounds/plugins/profilers/gpu_profile_decorator.py
+metaflow_extensions/outerbounds/plugins/profilers/simple_card_decorator.py
 metaflow_extensions/outerbounds/plugins/secrets/__init__.py
 metaflow_extensions/outerbounds/plugins/secrets/secrets.py
 metaflow_extensions/outerbounds/plugins/snowflake/__init__.py

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

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

{ob_metaflow_extensions-1.2.7rc1 → ob_metaflow_extensions-1.2.8}/setup.py

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

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

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