class_docstr: Context manager to register PyTree extension.
class_docstr_tok: ["Context","manager","to","register","PyTree","extension","."]
code: def __exit__(self, exc_type, exc_val, exc_tb):
      for class_type in self._extensions:
          pytree.SUPPORTED_NODES.pop(class_type)
  
imports: from __future__ import annotations
  import contextlib
  import functools
  import inspect
  from typing import Any, Callable, Dict, List, Mapping, Optional, Sequence, Tuple, Type, Union
  import torch._dynamo
  import torch.export
  import torch.fx
  import torch.onnx
  from torch.onnx._internal import _beartype, exporter, io_adapter
  from torch.utils import _pytree
label_desc: Use image node_id 3 for calling the _PyTreeExtensionContext obj's underlying member method code with example usage: obj.__exit__(exc_type, exc_val, exc_tb) without return types