Metadata-Version: 2.4
Name: sagemaker-mlops
Version: 1.1.1
Summary: SageMaker MLOps package for workflow orchestration and model building
Author: Amazon Web Services
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Requires-Python: >=3.9
Description-Content-Type: text/markdown
Requires-Dist: sagemaker-core>=2.1.1
Requires-Dist: sagemaker-train>=1.1.1
Requires-Dist: sagemaker-serve>=1.1.1
Requires-Dist: boto3<2.0,>=1.42.2
Requires-Dist: botocore<2.0,>=1.42.2
Provides-Extra: test
Requires-Dist: pytest; extra == "test"
Requires-Dist: pytest-cov; extra == "test"
Requires-Dist: mock; extra == "test"
Provides-Extra: dev
Requires-Dist: pytest; extra == "dev"
Requires-Dist: pytest-cov; extra == "dev"
Requires-Dist: mock; extra == "dev"
Requires-Dist: black; extra == "dev"
Requires-Dist: flake8; extra == "dev"

# SageMaker MLOps Package

The `sagemaker-mlops` package provides high-level orchestration capabilities for Amazon SageMaker workflows, including pipeline definitions, step implementations, and model building utilities.

## Purpose

This package sits at the top of the SageMaker SDK dependency hierarchy and orchestrates components from the Core, Train, and Serve packages. It resolves architectural violations by providing a dedicated home for workflow orchestration logic that needs to import from multiple lower-level packages.

## Architecture

The SageMaker SDK follows a clean 4-package architecture:

```
┌─────────────────────────────────────────────────────────────┐
│                   Package Architecture                       │
└─────────────────────────────────────────────────────────────┘

                    MLOps (orchestration)
                   ↙      ↓      ↘
              Train     Core     Serve
                   ↘      ↓      ↙
                       Core

Dependency Rules:
✓ Core → nothing (foundation layer)
✓ Train → Core only
✓ Serve → Core only
✓ MLOps → Train, Serve, Core (orchestration layer)
```

### Package Responsibilities

- **sagemaker-core**: Foundation primitives (entities, parameters, functions, conditions, properties)
- **sagemaker-train**: Training functionality (estimators, processors, tuners)
- **sagemaker-serve**: Serving functionality (models, predictors, endpoints)
- **sagemaker-mlops**: Workflow orchestration (pipelines, steps, model building)

## What's in This Package

### Workflow Orchestration (from Core)

The following files were moved from `sagemaker-core/src/sagemaker/core/workflow/` to establish clean architectural boundaries:

**Core Orchestration (2 files):**
- `pipeline.py` - Pipeline class for workflow definition
- `steps.py` - Base Step class and common step logic

**Step Implementations (13 files):**
- `automl_step.py` - AutoML training steps
- `model_step.py` - Model creation and registration steps
- `callback_step.py` - Callback steps for custom logic
- `clarify_check_step.py` - Model bias and explainability checks
- `condition_step.py` - Conditional execution steps
- `emr_step.py` - EMR cluster steps
- `fail_step.py` - Explicit failure steps
- `function_step.py` - Lambda function steps
- `lambda_step.py` - AWS Lambda invocation steps
- `monitor_batch_transform_step.py` - Batch transform monitoring
- `notebook_job_step.py` - Notebook execution steps
- `quality_check_step.py` - Model quality checks
- `step_collections.py` - Step collection utilities
- `step_outputs.py` - Step output handling

**Utilities (6 files):**
- `_utils.py` - Internal utilities
- `_steps_compiler.py` - Step compilation logic
- `_repack_model.py` - Model repackaging utilities
- `_event_bridge_client_helper.py` - EventBridge integration
- `triggers.py` - Pipeline triggers
- `utilities.py` - Public utility functions

**Configuration (6 files):**
- `check_job_config.py` - Quality check configuration
- `parallelism_config.py` - Parallel execution configuration
- `pipeline_definition_config.py` - Pipeline definition settings
- `pipeline_experiment_config.py` - Experiment tracking configuration
- `retry.py` - Retry policies
- `selective_execution_config.py` - Selective execution settings

### Model Building

ModelBuilder is now located in the `sagemaker-serve` package but is re-exported from MLOps for convenience.

### What Stayed in Core

The following primitive files remain in `sagemaker-core/src/sagemaker/core/workflow/`:

- `entities.py` - Base Entity and PipelineVariable classes
- `parameters.py` - Parameter type definitions
- `functions.py` - Pipeline functions (Join, JsonGet)
- `execution_variables.py` - ExecutionVariable
- `conditions.py` - Condition primitives
- `properties.py` - Property definitions
- `pipeline_context.py` - PipelineSession (refactored to remove Train/Serve imports)
- `__init__.py` - Package initialization

## Installation

Install the package in editable mode for development:

```bash
pip install -e sagemaker-mlops
```

Or install all SageMaker packages together:

```bash
pip install -e sagemaker-core
pip install -e sagemaker-train
pip install -e sagemaker-serve
pip install -e sagemaker-mlops
```
