diff --git a/pyproject.toml b/pyproject.toml index f5f22b06..a1898248 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -35,8 +35,15 @@ dev = [ "pre-commit>=3.6.0", ] -mcp = [ - "mcp>=0.1.0", +neural = [ + "torch>=2.0.0", + "torch-geometric>=2.3.0", + "tensorboard>=2.13.0", +] + +gnn = [ + "torch>=2.0.0", + "torch-geometric>=2.3.0", ] dashboard = [ @@ -44,13 +51,17 @@ dashboard = [ "dash-bootstrap-components>=1.5.0", ] +all = [ + "atomizer[neural,gnn,dev,dashboard]", +] + [build-system] requires = ["setuptools>=68.0", "wheel"] build-backend = "setuptools.build_meta" [tool.setuptools.packages.find] where = ["."] -include = ["mcp_server*", "optimization_engine*", "nx_journals*"] +include = ["optimization_engine*", "nx_journals*"] [tool.black] line-length = 100 @@ -73,4 +84,4 @@ testpaths = ["tests"] python_files = ["test_*.py"] python_classes = ["Test*"] python_functions = ["test_*"] -addopts = "-v --cov=mcp_server --cov=optimization_engine --cov-report=html --cov-report=term" +addopts = "-v --cov=optimization_engine --cov-report=html --cov-report=term"