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
Wrap an object into a FeatureStorage as specified by the ``register_storage_wrapper``
    decorators.
    
python/dgl/storages/base.py