Update

[ghstack-poisoned]
pytorch · Nov 18, 2024 · 4ed954f · 4ed954f
1 parent fef0365
commit 4ed954f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/test/test_transforms.py b/test/test_transforms.py
@@ -2076,7 +2076,10 @@ def test_transform_rb(self, rbclass):
         ):
             td = rb.sample(10)
 
+    @retry(AssertionError, tries=10, delay=0)
     def test_collector_match(self):
+        torch.manual_seed(0)
+
         # The counter in the collector should match the one from the transform
         t = TrajCounter()