Custom Templates

1from hopx_ai import Template
2
3# Build a custom template
4template = Template.build(
5    name="my-ml-environment",
6    base="code-interpreter",
7    
8    # Pre-install packages
9    setup_commands=[
10        "pip install torch torchvision transformers",
11        "pip install scikit-learn xgboost lightgbm",
12        "pip install mlflow wandb",
13    ],
14    
15    # Set environment variables
16    env_vars={
17        "CUDA_VISIBLE_DEVICES": "0",
18        "TRANSFORMERS_CACHE": "/workspace/.cache",
19    },
20    
21    # Copy files into template
22    files={
23        "/workspace/config.yaml": "model:
24  name: bert-base",
25    }
26)
27
28print(f"Template created: {template.id}")
29print(f"Status: {template.status}")
30
31# Use your custom template
32sandbox = Sandbox.create(template="my-ml-environment")