{
	"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"
			}
		}
	}
}