code: def wrap_storage(storage):
      """Wrap an object into a FeatureStorage as specified by the ``register_storage_wrapper``
      decorators.
      """
      for type_, storage_cls in STORAGE_WRAPPERS.items():
          if isinstance(storage, type_):
              return storage_cls(storage)
  
      assert isinstance(
          storage, FeatureStorage
      ), "The frame column must be a tensor or a FeatureStorage object, got {}".format(
          type(storage)
      )
      return storage
  
file_path: dgl/python/dgl/storages/base.py
imports: import threading
label_desc: Use image node_id 2 for calling a global function with example usage: wrap_storage(storage) and returns: storage, storage_cls