From 68bb70bbaef69c7e23a3d18a39c5375b5d511cc5 Mon Sep 17 00:00:00 2001 From: Dan Saunders Date: Thu, 21 Aug 2025 04:14:09 +0000 Subject: [PATCH] fix test --- tests/test_datasets.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_datasets.py b/tests/test_datasets.py index dcd3dc4d7..a29f155c8 100644 --- a/tests/test_datasets.py +++ b/tests/test_datasets.py @@ -673,12 +673,12 @@ class TestDatasetPreparation: cfg = DictDefault({"streaming": True, "eval_streaming": False}) 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 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") def test_eval_specific_mixing_configs(self):