Skip to content

Commit 5ce78df

Browse files
authored
Merge branch 'master' into conv1d_perf
2 parents 4ccfd7b + 20701e4 commit 5ce78df

35 files changed

Lines changed: 11667 additions & 126 deletions

.gitignore

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,3 +40,6 @@ tests/*.json
4040

4141
# Generated benchmark data (lives in gh-pages only)
4242
docs/benchmark/
43+
44+
# Local outputs from FlagTune
45+
/flagtune*/

pyproject.toml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -261,11 +261,15 @@ flaggems-setup = "flaggems_setup:main"
261261
# Package discovery (including shipping the tests/ and benchmark/ suites under
262262
# collision-safe names) is defined in setup.py, so it is intentionally omitted
263263
# here — declaring `packages` in both places is an error.
264+
flaggems-flagtune-pretune = "flag_gems.flagtune.cli.pretune:main"
265+
flaggems-flagtune-train = "flag_gems.flagtune.cli.train:main"
266+
flaggems-flagtune-compare = "flag_gems.flagtune.cli.compare:main"
264267

265268
[tool.setuptools.package-data]
266269

267270
"flag_gems" = ["backends.yaml"]
268271
"flag_gems.runtime" = ["*/**/*.yaml"]
272+
"flag_gems.flagtune.contracts" = ["configs/*.yaml"]
269273

270274
[tool.setuptools_scm]
271275
write_to = "src/flag_gems/_version.py"

src/flag_gems/flagtune/__init__.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
"""Offline FlagTune collection, training, and reporting for FlagGems.
2+
3+
This package is intentionally separate from :mod:`flag_gems.utils`: it owns a
4+
complete offline workflow rather than a reusable low-level helper. Import the
5+
library API from :mod:`flag_gems.flagtune.collection`; invoke user-facing
6+
commands through the three ``flaggems-flagtune-*`` console scripts.
7+
"""
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
"""Thin console-command entry points for FlagGems FlagTune workflows.
2+
3+
Each module owns argument parsing and process exit behavior for exactly one
4+
command: Pretune collection, ranker training, or run comparison.
5+
"""

0 commit comments

Comments
 (0)