Update

[ghstack-poisoned]
pytorch · Feb 10, 2025 · 0f8a761 · 0f8a761
2 parents e3a95dc + 5eb51d5
commit 0f8a761
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/test_env.py b/test/test_env.py
@@ -2572,6 +2572,7 @@ def main_collector(j, q=None):
             total_frames=N * n_workers * 100,
             storing_device=device,
             device=device,
+            trust_policy=True,
             cat_results=-1,
         )
         single_collectors = [
@@ -2581,6 +2582,7 @@ def main_collector(j, q=None):
                 frames_per_batch=n_workers * 100,
                 total_frames=N * n_workers * 100,
                 storing_device=device,
+                trust_policy=True,
                 device=device,
             )
             for i in range(n_workers)