remove iter multiplexed workers

karinazad · karinazad · commit 836f557affb8 · 2025-02-28T13:08:42.000-05:00
diff --git a/src/lobster/datasets/_multiplexed_sampling_dataset.py b/src/lobster/datasets/_multiplexed_sampling_dataset.py
@@ -4,7 +4,7 @@
 from torch.utils.data import IterableDataset
 
 
-class MultiplexedDataset(IterableDataset):
+class MultiplexedSamplingDataset(IterableDataset):
     """Dataset that samples from multiple datasets according to specified weights."""
 
     def __init__(
