Simple Workflow

from mixtrain import MixFlow


class ExampleWorkflow(MixFlow):
    """Example workflow with configurable inputs."""

    def setup(self):
        """Initialize the workflow."""
        print("Setting up workflow...")

    def run(
        self,
        model_name: str,
        learning_rate: float = 0.001,
        batch_size: int = 32,
        epochs: int = 10,
        use_gpu: bool = True,
    ):
        """Execute the workflow.

        Args:
            model_name: Name of the model to use (required)
            learning_rate: Learning rate for training
            batch_size: Training batch size
            epochs: Number of training epochs
            use_gpu: Whether to use GPU acceleration
        """
        print(f"Training {model_name} for {epochs} epochs")
        print(f"Batch size: {batch_size}")
        print(f"Learning rate: {learning_rate}")
        print(f"GPU: {'enabled' if use_gpu else 'disabled'}")

    def cleanup(self):
        """Clean up resources."""
        print("Cleaning up...")

mixtrain workflow create example_workflow.py \
  --name my-workflow \
  --description "Example workflow"

mixtrain workflow run my-workflow

mixtrain workflow run my-workflow \
  --input '{"model_name": "resnet50", "epochs": 20, "batch_size": 64}'