Skip to content

Pipeline config

Configuration dataclasses for AMLPipelineHelper

aml_connection_config dataclass

AML connection configuration

federated_config dataclass

Federated learning configuration

pipeline_cli_config dataclass

Pipeline config for command line parameters

pipeline_compute_config dataclass

AML workspace compute targets and I/O modes

silo dataclass

Silo info for federated learning

tenant_override_config dataclass

Tenant override configuration

default_config_dict()

Constructs the config dictionary for the pipeline helper settings

Source code in shrike/pipeline/pipeline_config.py
def default_config_dict():
    """Constructs the config dictionary for the pipeline helper settings"""
    return {
        "aml": aml_connection_config,
        "run": pipeline_cli_config,
        "compute": pipeline_compute_config,
        "module_loader": module_loader_config,
        "modules": module_manifest,
        "tenant_overrides": tenant_override_config,
        "federated_config": federated_config,
    }