pyproject.toml

[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"

[project]
name = "ambersim"
version = "0.0.1"
description = "In-house tools built on GPU-accelerated simulation"
readme = "README.md"
license = {text="MIT"}
requires-python = ">=3.8.0"
classifiers = [
    "Development Status :: 3 - Alpha",
    "Programming Language :: Python",
]
dependencies = [
    "brax>=0.9.3",
    "coacd>=1.0.0",
    "dm_control>=1.0.0",
    "flax>=0.7.5",
    "jax[cuda11_local]>=0.4.1",
    "jaxlib>=0.4.1",
    "matplotlib>=3.5.2",
    "mujoco>=3.0.0",
    "mujoco-mjx>=3.0.0",
    "numpy>=1.23.1",
    "scipy>=1.10.0",
    "torch>=1.13.1",
    "tensorboard>=2.15.1",
]

[project.optional-dependencies]

# Development packages
dev = [
    "black>=23.3.0",
    "flake8>=6.0.0",
    "isort>=5.12.0",
    "pre-commit>=3.3.2",
    "pyright>=1.1.332",
    "pytest>=7.4.2",
]

# Test-specific packages for verification
test = [
    "cvxpy>=1.4.1",
    "drake>=1.21.0",
    "libigl>=2.4.0",
    "pin>=2.6.20",
]

# All packages
all = ["ambersim[dev, test]"]

[project.scripts]
install-mujoco-from-src = "ambersim._scripts._install_shim:entrypoint"

[tool.setuptools.packages.find]
include = ["ambersim*"]

[tool.setuptools.package-data]
"*" = ["*.obj", "*.urdf", "*.xml", "install.sh", "install_mj_source.sh"]

[tool.black]
line-length = 120

[tool.pytest.ini_options]
testpaths = [
    "tests",
]

[tool.pyright]
include = ["ambersim"]
exclude = [
    "**/__pycache__",
]
defineConstant = { DEBUG = true }

reportMissingImports = "warning"
reportMissingTypeStubs = false
reportPrivateImportUsage = false

pythonVersion = "3.11"
pythonPlatform = "Linux"