-
Notifications
You must be signed in to change notification settings - Fork 5
Expand file tree
/
Copy pathpyproject.toml
More file actions
76 lines (67 loc) · 2.17 KB
/
pyproject.toml
File metadata and controls
76 lines (67 loc) · 2.17 KB
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
[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"
[project]
name = "sutro-yaro"
version = "0.1.0"
description = "Sutro Group research workspace for energy-efficient AI training"
readme = "README.md"
requires-python = ">=3.9"
license = {text = "MIT"}
authors = [
{name = "Sutro Group"},
]
keywords = ["sparse-parity", "energy-efficient-ml", "gymnasium", "reinforcement-learning"]
dependencies = [
"numpy",
]
[project.optional-dependencies]
# Gymnasium eval environment (SutroYaro/SparseParity-v0, SutroYaro/MultiChallenge-v0).
# `pip install -e .[eval]` gets you everything you need to run
# `gymnasium.make("SutroYaro/SparseParity-v0")` without any manual import.
eval = [
"gymnasium>=0.29",
]
# Modal backend for running the harness off-machine. Optional: only
# needed if you pass ``backend="modal"`` to the environment.
modal = [
"modal>=0.60",
]
# Developer / test tooling.
dev = [
"pytest>=7",
"pytest-xdist",
"ruff",
]
# Convenience "everything" extra for CI and contributors.
all = [
"gymnasium>=0.29",
"modal>=0.60",
"pytest>=7",
"pytest-xdist",
"ruff",
]
[project.urls]
Homepage = "https://github.com/cybertronai/SutroYaro"
Repository = "https://github.com/cybertronai/SutroYaro"
Documentation = "https://cybertronai.github.io/SutroYaro/"
# ---------------------------------------------------------------------------
# Gymnasium entry points
# ---------------------------------------------------------------------------
# Gymnasium discovers environments via the ``gymnasium.envs`` entry-point
# group (see https://gymnasium.farama.org/api/registry/). When gymnasium
# is imported it walks this group and imports every module listed, so the
# ``gymnasium.register()`` calls at module scope fire automatically.
#
# Result: after ``pip install -e .[eval]`` you can do
#
# import gymnasium as gym
# env = gym.make("SutroYaro/SparseParity-v0")
#
# with no prior ``import sparse_parity.eval``.
[project.entry-points."gymnasium.envs"]
SutroYaro = "sparse_parity.eval:register_all"
[tool.setuptools.packages.find]
where = ["src"]
[tool.setuptools.package-data]
"sparse_parity.eval" = ["answer_key.json", "README.md"]