pyproject.toml

[tool.poetry]
name = "cyberwheel"
version = "0.1.0"
description = "A simulation environment for training autonomous cyber defense agents based on the Mitre Att&ck framework."
authors = ["Sean Oesch <oeschts@ornl.gov>", "Cory Watson <watsoncl1@ornl.gov>", "Amul Chaulagain <chaulagaina@ornl.gov>", "Matthew Dixson <dixsonmk@ornl.gov>", "Brian Weber <weberb@ornl.gov>", "Phillipe Austria <austriaps@ornl.gov>"]
license = "MIT"
readme = "README.md"
repository = "https://github.com/ORNL/cyberwheel/"
classifiers = ["Private :: Do not Upload"]
packages = [{ include = "cyberwheel" }]

[tool.poetry.dependencies]
python = "~3.10"
networkx = "^3.2.1"
ipaddress = "^1.0.23"
pettingzoo = "^1.24.3"
pydantic = "^1"
jsonpickle = "^3.0.3"
wandb = "^0.16.5"
tensorboard = "^2.16.2"
tqdm = "^4.66.2"
ipykernel = "^6.29.4"
pygraphviz = "^1.13"
dash = "^2.17.0"
python-dotenv = "^1.0.1"
bigtree = "^0.19.3"
ray = "^2.38.0"
stable-baselines3 = "^2.3.2"
gymnasium = "0.28.1"


[tool.poetry.group.dev.dependencies]
black = "^24.2.0"
pre-commit = "^3.6.2"
pyright = "^1.1.351"
py-spy = "^0.3.14"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"

[tool.setuptools.packages.find]
where = ['cyberwheel']

[tool.setuptools.package-data]
cyberwheel = ['*.json', '*.yaml', '*.yml']