Vitessce

This tutorial has been adopted from the data preparation examples in the Vitessce documention.

Setup

Install dependencies:

pip install vitessce
pip install 'lamindb[jupyter,aws,bionty]'
!lamin load laminlabs/lamin-dev  # <-- replace with your instance
💡 connected lamindb: laminlabs/lamin-dev
from urllib.request import urlretrieve
from pathlib import Path
from anndata import read_h5ad
from vitessce import (
    VitessceConfig,
    Component as cm,
    AnnDataWrapper,
)
from vitessce.data_utils import (
    to_uint8,
    sort_var_axis,
    optimize_adata,
)
import lamindb as ln
/opt/hostedtoolcache/Python/3.10.14/x64/lib/python3.10/site-packages/vitessce/__init__.py:42: UserWarning: Extra installs are necessary to use widgets: No module named 'anywidget'
  warn(f'Extra installs are necessary to use widgets: {e}')
💡 connected lamindb: laminlabs/lamin-dev
# [optional] track the current notebook or script
ln.settings.transform.stem_uid = "BZhZQ6uIbkWv"
ln.settings.transform.version = "2.1"
ln.track()
💡 notebook imports: anndata==0.10.7 lamindb==0.74.0 vitessce==3.2.7
💡 loaded: Transform(uid='BZhZQ6uIbkWvFKml', version='2.1', name='Vitessce', key='vitessce', type='notebook', created_by_id=2, updated_at='2024-06-06 11:06:34 UTC')
💡 loaded: Run(uid='N1XzLjHKVkgil6Bd7BOC', started_at='2024-06-20 09:37:26 UTC', is_consecutive=True, transform_id=67, created_by_id=2)
Run(uid='N1XzLjHKVkgil6Bd7BOC', started_at='2024-06-20 09:37:26 UTC', is_consecutive=True, transform_id=67, created_by_id=2)

Pre-process the dataset

For this example, we use a dataset from the COVID-19 Cell Atlas https://www.covid19cellatlas.org/index.healthy.html#habib17.

# From https://github.com/vitessce/vitessce-python/blob/main/demos/habib-2017/src/convert_to_zarr.py
def convert_h5ad_to_zarr(input_path, output_path):
    adata = read_h5ad(input_path)
    adata = adata[:, adata.var["highly_variable"]].copy()
    leaf_list = sort_var_axis(adata.X, adata.var.index.values)
    adata = adata[:, leaf_list].copy()
    adata.layers["X_uint8"] = to_uint8(adata.X, norm_along="var")
    adata = optimize_adata(
        adata, obs_cols=["CellType"], obsm_keys=["X_umap"], layer_keys=["X_uint8"]
    )
    adata.write_zarr(output_path)
adata_filepath = "./habib17.processed.h5ad"
if not Path(adata_filepath).exists():
    urlretrieve(
        "https://covid19.cog.sanger.ac.uk/habib17.processed.h5ad", adata_filepath
    )
zarr_filepath = "./hhabib_2017_nature_methods.anndata.zarr"

convert_h5ad_to_zarr(adata_filepath, zarr_filepath)
Hide code cell output
/opt/hostedtoolcache/Python/3.10.14/x64/lib/python3.10/site-packages/anndata/compat/__init__.py:311: FutureWarning: Moving element from .uns['neighbors']['distances'] to .obsp['distances'].

This is where adjacency matrices should go now.
  warn(
/opt/hostedtoolcache/Python/3.10.14/x64/lib/python3.10/site-packages/anndata/compat/__init__.py:311: FutureWarning: Moving element from .uns['neighbors']['connectivities'] to .obsp['connectivities'].

This is where adjacency matrices should go now.
  warn(

Save dataset

zarr_artifact = ln.Artifact(
    zarr_filepath,
    description="Habib et al., 2017 Nature Methods, optimized anndata zarr",
)
zarr_artifact.save()
Artifact(uid='rcQQzTuhIGGa7zPljVIY', description='Habib et al., 2017 Nature Methods, optimized anndata zarr', suffix='.anndata.zarr', type='dataset', accessor='AnnData', size=15291221, hash='PUYrkaKY_2p2-UVr8HfJNg', hash_type='md5-d', n_objects=176, visibility=1, key_is_virtual=True, created_by_id=2, storage_id=1, transform_id=67, run_id=69, updated_at='2024-06-20 09:38:49 UTC')

Create a VitessceConfig object

Note

Here is a note on folder upload speed and we why chose to not use the .export(to="s3") functionality of Vitessce.

Set up the configuration by adding the views and datasets of interest.

vc = VitessceConfig(
    schema_version="1.0.15",
    description=zarr_artifact.description,
)
dataset = vc.add_dataset(name="Habib 2017").add_object(
    AnnDataWrapper(
        adata_url=zarr_artifact.path.to_url(),
        obs_feature_matrix_path="layers/X_uint8",
        obs_embedding_paths=["obsm/X_umap"],
        obs_embedding_names=["UMAP"],
        obs_set_paths=["obs/CellType"],
        obs_set_names=["Cell Type"],
    )
)
obs_sets = vc.add_view(cm.OBS_SETS, dataset=dataset)
obs_sets_sizes = vc.add_view(cm.OBS_SET_SIZES, dataset=dataset)
scatterplot = vc.add_view(cm.SCATTERPLOT, dataset=dataset, mapping="UMAP")
heatmap = vc.add_view(cm.HEATMAP, dataset=dataset)
genes = vc.add_view(cm.FEATURE_LIST, dataset=dataset)
vc.layout(((scatterplot | obs_sets) / heatmap) | (obs_sets_sizes / genes))

# inspect the config
vc.to_dict()
Hide code cell output
{'version': '1.0.15',
 'name': '',
 'description': 'Habib et al., 2017 Nature Methods, optimized anndata zarr',
 'datasets': [{'uid': 'A',
   'name': 'Habib 2017',
   'files': [{'fileType': 'anndata.zarr',
     'url': 'https://lamin-eu-central-1.s3-lamin-eu-central-1.amazonaws.com/9fm7UN13/.lamindb/rcQQzTuhIGGa7zPl.anndata.zarr',
     'options': {'obsEmbedding': [{'path': 'obsm/X_umap',
        'dims': [0, 1],
        'embeddingType': 'UMAP'}],
      'obsSets': [{'name': 'Cell Type', 'path': 'obs/CellType'}],
      'obsFeatureMatrix': {'path': 'layers/X_uint8'}}}]}],
 'coordinationSpace': {'dataset': {'A': 'A'}, 'embeddingType': {'A': 'UMAP'}},
 'layout': [{'component': 'obsSets',
   'coordinationScopes': {'dataset': 'A'},
   'x': 3.0,
   'y': 0.0,
   'w': 3.0,
   'h': 6.0},
  {'component': 'obsSetSizes',
   'coordinationScopes': {'dataset': 'A'},
   'x': 6.0,
   'y': 0.0,
   'w': 6.0,
   'h': 6.0},
  {'component': 'scatterplot',
   'coordinationScopes': {'dataset': 'A', 'embeddingType': 'A'},
   'x': 0.0,
   'y': 0.0,
   'w': 3.0,
   'h': 6.0},
  {'component': 'heatmap',
   'coordinationScopes': {'dataset': 'A'},
   'x': 0.0,
   'y': 6.0,
   'w': 6.0,
   'h': 6.0},
  {'component': 'featureList',
   'coordinationScopes': {'dataset': 'A'},
   'x': 6.0,
   'y': 6.0,
   'w': 6.0,
   'h': 6.0}],
 'initStrategy': 'auto'}

Save VitessceConfig object

from lamindb.integrations import save_vitessce_config
vc_artifact = save_vitessce_config(vc, description="View Habib17 in Vitessce")
💡 go to: https://lamin.ai/laminlabs/lamin-dev/artifact/i24hpBc8PtcZo17gZyuP
vc_artifact.view_lineage()
_images/ce50ea61646508eb9ed74979007d5378e6b3d8bea2d22279bf242e7805cfd256.svg
# [optional] save run report to share notebook with collaborators
# ln.finish()
Hide code cell content
# clean up artifacts in CI run
zarr_artifact.delete(permanent=True)
vc_artifact.delete(permanent=True)