fix missing __name__ attribute in partial function (#1466)

Summary:
This fixes the CI error introduced in the diff below this diff. Landing the fix here, will also see how this kicks off a PR in OSS repo.

The error was not caught in the CI of OSS because we might not have GPU code unit testing - need to fix that!

Differential Revision: D71564699

Author: divyanshk

torchdata/nodes/pin_memory.py

@@ -135,13 +135,16 @@ def reset(self, initial_state: Optional[Dict[str, Any]] = None):
         if self._it is not None:
             self._it._shutdown()
             del self._it
+
         self._it = _SingleThreadedMapper(
             source=self.source,
             prefetch_factor=1,
-            worker=functools.partial(
-                _pin_memory_loop,
-                device_id=self._current_device,
-                device=self._pin_memory_device,
+            worker=functools.wraps(_pin_memory_loop)(
+                functools.partial(
+                    _pin_memory_loop,
+                    device_id=self._current_device,
+                    device=self._pin_memory_device,
+                )
             ),
             snapshot_frequency=self.snapshot_frequency,
             initial_state=initial_state,