sagemaker.core.telemetry.attribution

sagemaker.core.telemetry.attribution#

Attribution module for tracking the provenance of SDK usage.

Functions

set_attribution(attribution)

Sets the SDK usage attribution to the specified source.

Classes

Attribution(value)

Enumeration of known SDK attribution sources.
class sagemaker.core.telemetry.attribution.Attribution(value)[source]#

Bases: Enum

Enumeration of known SDK attribution sources.

SAGEMAKER_AGENT_PLUGIN = 'awslabs/agent-plugins/sagemaker-ai'#
sagemaker.core.telemetry.attribution.set_attribution(attribution: Attribution)[source]#

Sets the SDK usage attribution to the specified source.

Call this at the top of scripts generated by an agent or integration to enable accurate telemetry attribution.

Parameters:

attribution (Attribution) – The attribution source to set.

Raises:

TypeError – If attribution is not an Attribution enum member.

Contents