File size: 819 Bytes
925ee3b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 | """Pooch-based download registry for scPTR datasets."""
from __future__ import annotations
from pathlib import Path
import pooch
_CACHE_DIR = Path.home() / ".cache" / "scptr"
# Base URL for scvelo's dataset hosting on GitHub
_SCVELO_BASE = "https://github.com/theislab/scvelo_notebooks/raw/master/"
REGISTRY = pooch.create(
path=_CACHE_DIR,
base_url="",
registry={
"pancreas.h5ad": None,
"dentate_gyrus.h5ad": None,
},
urls={
"pancreas.h5ad": _SCVELO_BASE + "data/Pancreas/endocrinogenesis_day15.h5ad",
"dentate_gyrus.h5ad": _SCVELO_BASE + "data/DentateGyrus/10X43_1.h5ad",
},
)
def fetch(name: str) -> str:
"""Fetch a dataset file, downloading if necessary.
Returns the local file path.
"""
return REGISTRY.fetch(name, progressbar=True)
|