tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix test

Browse Source
This commit is contained in:
Dan Saunders
2025-08-21 04:14:09 +00:00
parent 5d8d7ef327
commit 68bb70bbae
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_datasets.py
View File

@@ -673,12 +673,12 @@ class TestDatasetPreparation:
cfg = DictDefault({"streaming": True, "eval_streaming": False}) cfg = DictDefault({"streaming": True, "eval_streaming": False})
assert _is_streaming_enabled_for_split(cfg, "train") assert _is_streaming_enabled_for_split(cfg, "train")
assert _is_streaming_enabled_for_split(cfg, "test") assert not _is_streaming_enabled_for_split(cfg, "test")
# Test train streaming disabled, eval streaming enabled # Test train streaming disabled, eval streaming enabled
cfg2 = DictDefault({"streaming": False, "eval_streaming": True}) cfg2 = DictDefault({"streaming": False, "eval_streaming": True})
assert _is_streaming_enabled_for_split(cfg2, "train") assert not _is_streaming_enabled_for_split(cfg2, "train")
assert _is_streaming_enabled_for_split(cfg2, "test") assert _is_streaming_enabled_for_split(cfg2, "test")
def test_eval_specific_mixing_configs(self): def test_eval_specific_mixing_configs(self):