From 25062f22d8fcb4f34d7d0a8b077e0bde23ecaf97 Mon Sep 17 00:00:00 2001
From: Divyansh Khanna <divyanshkhanna@meta.com>
Date: Tue, 21 Jan 2025 10:36:55 -0800
Subject: [PATCH] test for test flakiness

---
 test/nodes/test_multi_node_weighted_sampler.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/nodes/test_multi_node_weighted_sampler.py b/test/nodes/test_multi_node_weighted_sampler.py
index 978919d24..b9f00d926 100644
--- a/test/nodes/test_multi_node_weighted_sampler.py
+++ b/test/nodes/test_multi_node_weighted_sampler.py
@@ -43,7 +43,7 @@ def _setup_multi_node_weighted_sampler(self, num_samples, num_datasets, weights_
         datasets = {f"ds{i}": IterableWrapper(DummyIterableDataset(num_samples, f"ds{i}")) for i in range(num_datasets)}
         weights = {f"ds{i}": weights_fn(i) for i in range(num_datasets)}
         node = MultiNodeWeightedSampler(datasets, weights, stop_criteria)
-        return Prefetcher(node, prefetch_factor=1)
+        return Prefetcher(node, prefetch_factor=3)
 
     def test_multi_node_weighted_sampler_weight_sampler_keys_mismatch(self) -> None:
         """Test validation logic for MultiNodeWeightedSampler if the keys of source_nodes and weights are not the same"""