automatically split out reasoning trace from dataset (#2579)

* automatically split out reasoning trace from dataset * chore: lint * fix import
2025-04-28 18:23:03 -04:00
parent 63b17e3109
commit 2d77165dc0
4 changed files with 144 additions and 0 deletions
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -90,6 +90,12 @@ def download_qwen_2_5_half_billion_model():
    snapshot_download_w_retry("Qwen/Qwen2.5-0.5B", repo_type="model")


+@pytest.fixture(scope="session", autouse=True)
+def download_qwen3_half_billion_model():
+    # download the model
+    snapshot_download_w_retry("Qwen/Qwen3-0.6B", repo_type="model")
+
+
@pytest.fixture(scope="session", autouse=True)
 def download_tatsu_lab_alpaca_dataset():
    # download the dataset