Skip to content

Commit e37a524

Browse files
authored
fix: remove dead warnings (#520)
1 parent 23b05ed commit e37a524

File tree

2 files changed

+0
-10
lines changed

2 files changed

+0
-10
lines changed

src/litdata/streaming/dataset.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -112,12 +112,6 @@ def __init__(
112112
"Please provide a valid ParquetLoader instance."
113113
)
114114

115-
if item_loader is not None and item_loader._low_memory and shuffle:
116-
raise ValueError(
117-
"You have enabled shuffling when using low memory with ParquetLoader. "
118-
"This configuration may lead to performance issues during the training process. "
119-
"Consider disabling shuffling or using a ParquetLoader without low memory mode."
120-
)
121115
item_loader = item_loader or ParquetLoader()
122116

123117
self.input_dir = input_dir

tests/streaming/test_parquet.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,3 @@ def test_stream_hf_parquet_dataset(monkeypatch, huggingface_hub_fs_mock, pq_data
204204
assert _ds[0] == pq_data["name"][idx]
205205
assert _ds[1] == pq_data["weight"][idx]
206206
assert _ds[2] == pq_data["height"][idx]
207-
208-
# Test case 5: Streaming with ParquetLoader and low_memory=True and shuffle=True
209-
with pytest.raises(ValueError, match="You have enabled shuffling when using low memory with ParquetLoader."):
210-
StreamingDataset(hf_url, item_loader=ParquetLoader(low_memory=True), shuffle=True)

0 commit comments

Comments
 (0)