pyproject.toml

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project]
name = "ms_hgnn"
version = "0.1.0"
dependencies = [
  "numpy==1.26.4",
  "matplotlib==3.8.3",
  "networkx==3.2.1",
  "urchin==0.0.27",
  "torch==2.2.1",
  "torchvision==0.17.1",
  "torchmetrics==1.3.2",
  "torch_geometric==2.5.0",
  "pandas==2.2.1",
  "rospkg>=1.5.0",
  "rosbags==0.9.22",
  "lightning==2.2.1",
  "wandb==0.16.6",
  "scipy==1.12.0",
  "names==0.3.0",
  "scikit-learn==1.4.1.post1",
  "gdown==5.2.0"
]
[tool.hatch.build.targets.wheel]
packages = ["src/ms_hgnn"]
authors = [
  { name="Fengze Xie", email="fxxie@caltech.edu"},
  { name="Sizhe Wei", email="swei@gatech.edu" },
  { name="Lu Gan", email="lgan@gatech.edu" },
]
description = "Morphological-Symmetry-Equivariant Heterogeneous Graph Neural Network for Robotic Dynamics Learning (MS-HGNN) for robotic dynamics learning."
readme = "README.md"
requires-python = ">=3.11"
classifiers = [
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.11",
    "Intended Audience :: Science/Research",
    "Operating System :: POSIX :: Linux",
]

[project.urls]
Homepage = "https://lunarlab-gatech.github.io/MorphSym-HGNN/"
Issues = ""