{

    "name": "BioNeMo2 Development Container",

    "runArgs": [

        "--gpus=all",

        "--shm-size=4g"

    ],

    "build": {

        "context": "${localWorkspaceFolder}",

        "dockerfile": "${localWorkspaceFolder}/Dockerfile",

        "cacheFrom": "type=registry,ref=nvcr.io/nvidian/cvai_bnmo_trng/bionemo:bionemo2-devcontainer-cache",

        "cacheTo": "type=registry,ref=nvcr.io/nvidian/cvai_bnmo_trng/bionemo:bionemo2-devcontainer-cache,mode=max",

        "target": "dev"

    },

    "mounts": [

        // Mount the local ~/.aws config to pass along AWS credentials for PBSS.

        "source=${localEnv:HOME}/.aws,target=/home/ubuntu/.aws,type=bind,consistency=cached",

        "source=${localEnv:HOME}/.ngc,target=/home/ubuntu/.ngc,type=bind,consistency=cached",

        "source=${localEnv:HOME}/.cache,target=/home/ubuntu/.cache,type=bind,consistency=cached",

        "source=${localEnv:HOME}/.ssh,target=/home/ubuntu/.ssh,readonly,type=bind,consistency=cached",

        "source=${localEnv:HOME}/.netrc,target=/home/ubuntu/.netrc,readonly,type=bind,consistency=cached",

        // Mount bash history file for persistence. Created if it doesn't exist in initializeCommand.sh

        "source=${localEnv:HOME}/.bash_history_devcontainer,target=/home/ubuntu/.bash_history,type=bind,consistency=cached"

    ],

    "containerEnv": {

        "TMPDIR": "/tmp",

        "NUMBA_CACHE_DIR": "/tmp/"

    },

    "postCreateCommand": "./.devcontainer/postCreateCommand.sh",

    "initializeCommand": "./.devcontainer/initializeCommand.sh",

    "remoteUser": "ubuntu",

    "customizations": {

        "vscode": {

            "extensions": [

                "ms-python.python",

                "ms-python.vscode-pylance",

                "eamodio.gitlens",

                "streetsidesoftware.code-spell-checker",

                "ms-azuretools.vscode-docker",

                "charliermarsh.ruff",

                "njpwerner.autodocstring",

                "ms-toolsai.jupyter",

                "tamasfe.even-better-toml"

            ],

            "settings": {

                "python.analysis.extraPaths": [

                    "./sub-packages/bionemo-fw/src",

                    "./sub-packages/bionemo-core/src",

                    "./sub-packages/bionemo-esm2/src",

                    "./sub-packages/bionemo-geneformer/src",

                    "./sub-packages/bionemo-llm/src",

                    "./sub-packages/bionemo-testing/src",

                    "./sub-packages/bionemo-amplify/src",

                    "./sub-packages/bionemo-example_model/src",

                    "./3rdparty/NeMo",

                    "./3rdparty/Megatron-LM"

                ],

                "python.defaultInterpreterPath": "/usr/bin/python",

                "python.testing.pytestEnabled": true,

                "python.testing.pytestArgs": [

                    "sub-packages/",

                    "scripts/"

                ],

                "python.analysis.typeCheckingMode": "standard"

            }

        }

    }

}