Skip to content

Commit bf376b1

Browse files
committed
move lean progress into lean_dojo_v2
1 parent 23c71eb commit bf376b1

6 files changed

Lines changed: 5 additions & 3 deletions

File tree

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -231,11 +231,11 @@ The generated artifacts flow into the `DynamicDatabase`, which keeps repositorie
231231

232232
- Generate a JSONL dataset with remaining-step targets (or replace it with your own LeanProgress export):
233233
```sh
234-
python examples/leanprogress/create_sample_dataset.py --output raid/data/sample_leanprogress_dataset.jsonl
234+
python -m lean_dojo_v2.lean_progress.create_sample_dataset --output raid/data/sample_leanprogress_dataset.jsonl
235235
```
236236
- Fine-tune a regression head that predicts `steps_remaining`:
237237
```sh
238-
python examples/leanprogress/train_steps_model.py \
238+
python -m lean_dojo_v2.lean_progress.train_steps_model \
239239
--dataset raid/data/sample_leanprogress_dataset.jsonl \
240240
--output-dir raid/checkpoints/leanprogress_steps \
241241
--model-name bert-base-uncased

examples/leanprogress/create_sample_dataset.py renamed to lean_dojo_v2/lean_progress/create_sample_dataset.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,3 +69,4 @@ def main() -> None:
6969

7070
if __name__ == "__main__":
7171
main()
72+

examples/leanprogress/train_steps_model.py renamed to lean_dojo_v2/lean_progress/train_steps_model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -150,3 +150,4 @@ def main() -> None:
150150

151151
if __name__ == "__main__":
152152
main()
153+
File renamed without changes.

tests/test_leanprogress_examples.py renamed to lean_dojo_v2/tests/test_leanprogress_examples.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import json
22
from types import SimpleNamespace
33

4-
from examples.leanprogress import create_sample_dataset, train_steps_model
4+
from lean_dojo_v2.lean_progress import create_sample_dataset, train_steps_model
55

66

77
def test_create_sample_dataset_writes_expected_jsonl(tmp_path, monkeypatch):

0 commit comments

Comments
 (0)