arla/pyproject.toml at main · renbytes/arla · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
[tool.poetry]
name = "arla"
version = "0.1.0"
description = "A modular, scalable framework for running multi-agent simulations with affective reflective learning."
authors = ["Bordumb <bordumbb@gmail.com>"]
readme = "README.md"
homepage = "https://github.com/renbytes/arla"
repository = "https://github.com/renbytes/arla"
documentation = "https://github.com/renbytes/arla"

# Expose all packages from their src/ directories
[[tool.poetry.packages]]
include = "agent_sim"
from = "agent-sim/src"

[[tool.poetry.packages]]
include = "agent_core"
from = "agent-core/src"

[[tool.poetry.packages]]
include = "agent_engine"
from = "agent-engine/src"

[[tool.poetry.packages]]
include = "agent_persist"
from = "agent-persist/src"

[[tool.poetry.packages]]
include = "agent_concurrent"
from = "agent-concurrent/src"

[tool.poetry.scripts]
agentsim = "agent_sim.cli:app"
arla = "agent_sim.cli:app"

[tool.poetry.dependencies]
python = "==3.11.9"

# Local path dependencies (editable)
agent-core = { path = "agent-core", develop = true }
agent-engine = { path = "agent-engine", develop = true }
agent-concurrent = { path = "agent-concurrent", develop = true }
agent-persist = { path = "agent-persist", develop = true }
agent-sim = { path = "agent-sim", develop = true }

# Core infrastructure
celery = { extras = ["redis"], version = "==5.3.4" }
redis = ">=4.3.0,<7.0.0"
hydra-core = ">=1.3.2,<2.0.0"
omegaconf = ">=2.3.0,<3.0.0"

# Database
sqlalchemy = ">=2.0.0,<3.0.0"
greenlet = ">=3.0.0,<4.0.0"
alembic = ">=1.8.0,<2.0.0"
asyncpg = ">=0.25.0,<1.0.0"
psycopg2-binary = ">=2.9.0,<3.0.0"

# Data validation and models
pydantic = ">=2.0.0,<3.0.0"

# Data science stack
pandas = ">=1.4.0,<3.0.0"
numpy = ">=1.21.0,<3.0.0"
scipy = ">=1.8.0,<2.0.0"
scikit-learn = ">=1.1.0,<2.0.0"
torch = ">=2.0.0,<3.0.0"

# Visualization
matplotlib = ">=3.5.0,<4.0.0"
seaborn = ">=0.11.0,<1.0.0"
imageio = ">=2.19.0,<3.0.0"

# Network analysis and causal inference
networkx = ">=3.0.0,<4.0.0"
dowhy = ">=0.8.0,<1.0.0"
pygraphviz = "1.13"

# API and web
uvicorn = ">=0.18.0,<1.0.0"
fastapi = ">=0.85.0,<1.0.0"

# ML tracking and experiment management
mlflow = ">=2.0.0,<4.0.0"

# AI and language models
openai = ">=1.0.0,<2.0.0"

# CLI and UI
rich = ">=13.0.0,<15.0.0"
typer = ">=0.9.0,<1.0.0"

# Utilities
python-dotenv = ">=0.19.0,<2.0.0"

[tool.poetry.group.dev.dependencies]
# Testing
pytest = ">=7.0.0,<9.0.0"
pytest-mock = ">=3.10.0,<4.0.0"
pytest-asyncio = ">=0.21.0,<1.0.0"
pytest-cov = ">=4.0.0,<6.0.0"
coverage = { extras = ["toml"], version = ">=7.0.0,<8.0.0" }

# Code quality
ruff = "==0.12.3"
mypy = "==1.16.1"
pre-commit = ">=3.0.0,<5.0.0"

# Development tools
flower = ">=2.0.0,<3.0.0"

[tool.poetry.group.docs.dependencies]
mkdocs = ">=1.5.0,<2.0.0"
mkdocs-material = ">=9.0.0,<10.0.0"
mkdocs-rss-plugin = ">=1.12.0,<2.0.0"
mkdocs-autorefs = ">=0.5.0,<1.0.0"
mkdocstrings = { extras = ["python"], version = ">=0.24.0,<1.0.0" }

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

[tool.mypy]
# Tell mypy to treat directories with __init__.py as the root of a package
explicit_package_bases = true
namespace_packages = true
exclude = "repositories\\.arla\\..*"
disallow_untyped_defs = true
# Define the specific paths mypy should check
files = [
    "agent-core/src",
    "agent-engine/src",
    "agent-sim/src",
    "agent-concurrent/src",
    "agent-persist/src",
    "simulations",
    "scripts"
]
# Standard quality-of-life settings
ignore_missing_imports = true
follow_imports = "silent"