switch from print to logging

Kai-46 · Kai-46 · commit dba2ebaa6598 · 2025-07-14T18:34:58.000-07:00
diff --git a/torch/utils/data/_utils/fetch.py b/torch/utils/data/_utils/fetch.py
@@ -5,8 +5,12 @@
 """
 
 import asyncio
+import logging
 from concurrent.futures import ThreadPoolExecutor
 
+# Set up logger for this module
+logger = logging.getLogger(__name__)
+
 
 class _BaseDatasetFetcher:
     def __init__(self, dataset, auto_collation, collate_fn, drop_last, num_threads=1):
@@ -31,7 +35,11 @@ async def _worker(task_queue, result_queue):
                     )
                     await result_queue.put((index, result))
                 except Exception as e:
-                    print(f"Exception during fetch: {e}")
+                    logger.error(
+                        "Exception during fetch for index %s: %s (%s)", 
+                        index, str(e), type(e).__name__,
+                        exc_info=True
+                    )
                 finally:
                     task_queue.task_done()
 
