| 1 | // AWS-owned SageMaker scikit-learn serving images (per-region registries). |
| 2 | // CreateModel performs a real pull-permission check, so test models must |
| 3 | // reference an image that actually exists and that the execution role (with |
| 4 | // AmazonEC2ContainerRegistryReadOnly) can pull. |
| 5 | export const sklearnImage = (region: string) => { |
| 6 | const registries: Record<string, string> = { |
| 7 | "us-east-1": "683313688378", |
| 8 | "us-east-2": "257758044811", |
| 9 | "us-west-1": "746614075791", |
| 10 | "us-west-2": "246618743249", |
| 11 | }; |
| 12 | const registry = registries[region]; |
| 13 | if (!registry) { |
| 14 | throw new Error(`no sklearn registry mapped for region ${region}`); |
| 15 | } |
| 16 | return `${registry}.dkr.ecr.${region}.amazonaws.com/sagemaker-scikit-learn:1.2-1`; |
| 17 | }; |