code: def pygfunc(func):
      r"""A decorator for pyg transforms. You may want to use it to quickly wrap a feature transform function object.
      Examples
      --------
      @register_pyg
      @pygfunc(local_degree_profile)
      class PYGLocalDegreeProfile(local_degree_profile):pass
      """
  
      def decorator_func(cls):
          cls.extract = lambda s, data: (
              func(*s._args, **s._kwargs)(data)
          ).x
          return cls
  
      return decorator_func
  
imports: from torch_geometric.transforms.one_hot_degree import OneHotDegree
  from torch_geometric.utils import degree
  from torch_geometric.transforms.normalize_features import NormalizeFeatures
  from torch_geometric.transforms.local_degree_profile import LocalDegreeProfile
  from ..None import register_feature
  from .base import BaseGenerator
  import numpy
  from ..None import register_feature
  import torch
  from functools import wraps
label_desc: Use image node_id 2 for calling a global function with example usage: pygfunc(func) and returns: decorator_func, cls