File size: 1,421 Bytes
5f78183
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
[build-system]
requires = ["setuptools>=45", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "openenv-cernenv"
version = "0.1.0"
description = "RL environment for autonomous particle physics agents at the LHC"
requires-python = ">=3.10,<3.13"
dependencies = [
    "openenv-core[core]>=0.2.0",
    "numpy>=1.24.0",
    "scipy>=1.10.0",
    "pydantic>=2.0.0",
]

[project.optional-dependencies]
dev = [
    "pytest>=8.0.0",
    "pytest-cov>=4.0.0",
]
train = [
    "accelerate>=1.0.0",
    "datasets>=2.18.0",
    "matplotlib>=3.8.0",
    "peft>=0.10.0",
    "torch>=2.2.0",
    "transformers>=4.44.0",
    "trl>=0.9.0",
]

[project.scripts]
cernenv-server = "server.app:main"

[tool.uv]
package = false

[tool.setuptools]
include-package-data = true
packages = [
    "cernenv",
    "cernenv.server",
    "cernenv.server.simulator",
    "cernenv.server.rules",
    "cernenv.server.rewards",
    "cernenv.server.tasks",
    "cernenv.server.physics",
    "cernenv.training",
    "cernenv.tests",
]

[tool.setuptools.package-dir]
cernenv = "."
"cernenv.server" = "server"
"cernenv.server.simulator" = "server/simulator"
"cernenv.server.rules" = "server/rules"
"cernenv.server.rewards" = "server/rewards"
"cernenv.server.tasks" = "server/tasks"
"cernenv.server.physics" = "server/physics"
"cernenv.training" = "training"
"cernenv.tests" = "tests"