fix(profiling): call restore() in PipelineProfiler.run() after profiling

Store the restore callable from annotate_pipeline() and call it at
the end of run() so class-level method patches are cleaned up.

Author: Tai An

examples/profiling/profiling_utils.py

@@ -169,7 +169,9 @@ def setup_pipeline(self, annotate=True):
         pipe.set_progress_bar_config(disable=True)
 
         if annotate:
-            annotate_pipeline(pipe)
+            self._restore_annotations = annotate_pipeline(pipe)
+        else:
+            self._restore_annotations = None
         return pipe
 
     def run(self):
@@ -215,7 +217,9 @@ def run(self):
             )
         )
 
-        # Cleanup
+        # Cleanup -- restore patched methods so class-level patches don't persist
+        if self._restore_annotations is not None:
+            self._restore_annotations()
         pipe.to("cpu")
         del pipe
         flush()