remove prints from test_map (#1347)

pytorch · Oct 25, 2024 · dcaa6ea · dcaa6ea
1 parent a087018
commit dcaa6ea
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/test/nodes/test_map.py b/test/nodes/test_map.py
@@ -36,7 +36,7 @@ def _test_exception_handling_mapper(self, pin_memory, method):
         node = Prefetcher(node, prefetch_factor=2)
 
         with self.assertRaisesRegex(ValueError, "test exception"):
-            print(list(node))
+            list(node)
 
     def test_exception_handling_mapper(self):
         self._test_exception_handling_mapper(False, "thread")
@@ -71,7 +71,6 @@ def _test_map(self, in_order, method) -> None:
         results: List[List[dict]] = [[], []]
         for epoch in range(2):
             for batch in node:
-                print(f"{epoch=}, {batch=}")
                 results[epoch].extend(batch)
 
         for result in results: