image.def

Bootstrap: docker
From: ghcr.io/prefix-dev/pixi:latest
Stage: build

%arguments
    CUDA_VERSION=12.5
    ENVIRONMENT=nsbi-env-gpu

%files
    ./pixi.toml /app/
    ./pixi.lock /app/
    ./README.md /app/
    ./src /app/ 
    ./pyproject.toml /app/

%post
    #!/bin/bash
    export CONDA_OVERRIDE_CUDA={{ CUDA_VERSION }}
    export SETUPTOOLS_SCM_PRETEND_VERSION_FOR_NSBI_COMMON_UTILS=0.0.0
    export SETUPTOOLS_SCM_PRETEND_VERSION=0.0.0

    set -e
    cd /app
    pixi install --environment {{ ENVIRONMENT }}

    # Build an entrypoint that activates the env then runs the command
    echo "#!/bin/bash" > /app/entrypoint.sh
    pixi shell-hook --environment {{ ENVIRONMENT }} -s bash >> /app/entrypoint.sh
    echo 'exec "$@"' >> /app/entrypoint.sh
    chmod +x /app/entrypoint.sh


Bootstrap: docker
From: ubuntu:24.04
Stage: final

%arguments
    ENVIRONMENT=nsbi-env-gpu

%files from build
    /app/.pixi/envs/{{ ENVIRONMENT }} /app/.pixi/envs/{{ ENVIRONMENT }}
    /app/pixi.toml /app/pixi.toml
    /app/pixi.lock /app/pixi.lock
    /app/.pixi/.condapackageignore /app/.pixi/.condapackageignore

%environment
    export PATH=/app/.pixi/envs/{{ ENVIRONMENT }}/bin:$PATH

%runscript
    #!/bin/bash
    /app/entrypoint.sh "$@"