ml-pipeline
Warn
Audited by Gen Agent Trust Hub on Aug 27, 2026
Risk Level: MEDIUMDYNAMIC_EXECUTIONINDIRECT_PROMPT_INJECTION
Full Analysis
- [DYNAMIC_EXECUTION]: The skill uses
pickle.load(),joblib.load(), andtorch.load()to deserialize models and training artifacts. These methods are inherently unsafe when used with untrusted data, as they can execute arbitrary code during the loading process. - Evidence in
references/training-pipelines.md:torch.load(checkpoint_path / "checkpoint.pt", ...) - Evidence in
references/pipeline-orchestration.md:joblib.load(model_artifact.path) - Evidence in
references/feature-engineering.md:pickle.load(f)in theFeaturePipeline.loadmethod. - [INDIRECT_PROMPT_INJECTION]: The skill facilitates the ingestion of external data from various sources (local files, S3, GCS) for training and feature engineering. While typical for ML workflows, this creates an attack surface where maliciously crafted data files could potentially influence the agent's behavior if instructions were embedded within the data.
- Ingestion points: The
data_pathparameter inml_pipeline(references/pipeline-orchestration.md) and variousload_datatasks serve as entry points for untrusted data. - Boundary markers: The skill does not implement specific delimiters or instructions to ignore potential commands embedded within data files.
- Capability inventory: The skill templates utilize file system writes, subprocess execution (via containerized orchestration components), and network operations to interact with cloud storage (S3, GCS).
- Sanitization: Data is validated for schema and statistical distribution using
great_expectations, but there is no sanitization of content against adversarial prompts or embedded instructions.
Audit Metadata