zmap-tools 0.2.2.2__tar.gz → 0.2.2.3__tar.gz

{zmap_tools-0.2.2.2 → zmap_tools-0.2.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zmap-tools
-Version: 0.2.2.2
+Version: 0.2.2.3
 Summary: API for working with ZMAP (Zebrafish Multi-Atlas Project)
 Author-email: Daniel Wagner <Daniel.Wagner@ucsf.edu>
 License: BSD 3-Clause License
@@ -44,26 +44,30 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: anndata==0.12.18
-Requires-Dist: scanpy==1.12.1
-Requires-Dist: numpy==2.0.2
-Requires-Dist: numba==0.60.0
-Requires-Dist: llvmlite==0.43.0
-Requires-Dist: pandas<3.0,>=2.0
-Requires-Dist: matplotlib<4.0,>=3.7
+Requires-Dist: anndata<0.14,>=0.12.18
+Requires-Dist: scanpy<1.13,>=1.12.1
+Requires-Dist: numpy<3,>=2.1
+Requires-Dist: numba>=0.64
+Requires-Dist: pandas<3,>=2.2
+Requires-Dist: matplotlib<4,>=3.7
 Requires-Dist: requests>=2.31
-Requires-Dist: scipy==1.16.3
-Requires-Dist: scikit-learn==1.6.1
-Requires-Dist: faiss-cpu==1.14.3
-Requires-Dist: seaborn==0.13.2
+Requires-Dist: scipy<2,>=1.15
+Requires-Dist: scikit-learn<2,>=1.6
+Requires-Dist: faiss-cpu<2,>=1.14.3
+Requires-Dist: seaborn<0.14,>=0.13.2
 Requires-Dist: tqdm>=4.66
 Requires-Dist: adjustText>=0.8
-Requires-Dist: symphonypy==0.2.4
-Requires-Dist: igraph==1.0.0
-Requires-Dist: leidenalg==0.12.0
+Requires-Dist: symphonypy<0.3,>=0.2.4
+Requires-Dist: igraph<2,>=1.0
+Requires-Dist: leidenalg<0.13,>=0.12
+Requires-Dist: joblib>=1.3
+Requires-Dist: h5py>=3.0
+Requires-Dist: torch>=2.0
+Requires-Dist: lightgbm>=4.0
 Dynamic: license-file
 
 # zmap-tools

{zmap_tools-0.2.2.2 → zmap_tools-0.2.2.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "zmap-tools"
-version = "0.2.2.2"
+version = "0.2.2.3"
 description = "API for working with ZMAP (Zebrafish Multi-Atlas Project)"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -24,28 +24,32 @@ classifiers = [
   "Programming Language :: Python :: 3.10",
   "Programming Language :: Python :: 3.11",
   "Programming Language :: Python :: 3.12",
+  "Programming Language :: Python :: 3.13",
 ]
 
 dependencies = [
-  "anndata==0.12.18",
-  "scanpy==1.12.1",
+  "anndata>=0.12.18,<0.14",
+  "scanpy>=1.12.1,<1.13",
 
-  "numpy==2.0.2",
-  "numba==0.60.0",
-  "llvmlite==0.43.0",
+  "numpy>=2.1,<3",
+  "numba>=0.64",
 
-  "pandas>=2.0,<3.0",
-  "matplotlib>=3.7,<4.0",
+  "pandas>=2.2,<3",
+  "matplotlib>=3.7,<4",
   "requests>=2.31",
-  "scipy==1.16.3",
-  "scikit-learn==1.6.1",
-  "faiss-cpu==1.14.3",
-  "seaborn==0.13.2",
+  "scipy>=1.15,<2",
+  "scikit-learn>=1.6,<2",
+  "faiss-cpu>=1.14.3,<2",
+  "seaborn>=0.13.2,<0.14",
   "tqdm>=4.66",
   "adjustText>=0.8",
-  "symphonypy==0.2.4",
-  "igraph==1.0.0",
-  "leidenalg==0.12.0",
+  "symphonypy>=0.2.4,<0.3",
+  "igraph>=1.0,<2",
+  "leidenalg>=0.12,<0.13",
+  "joblib>=1.3",
+  "h5py>=3.0",
+  "torch>=2.0",
+  "lightgbm>=4.0",
 ]
 
 [project.urls]
@@ -55,6 +59,12 @@ Issues = "https://github.com/WagnerLabUCSF/zmap-tools/issues"
 
 [tool.setuptools]
 package-dir = {"" = "src"}
+include-package-data = true
 
 [tool.setuptools.packages.find]
 where = ["src"]
+
+[tool.setuptools.package-data]
+zmap = [
+  "predict/baseline2_file/**/*",
+]

zmap_tools-0.2.2.3/src/zmap/__init__.py

@@ -0,0 +1,76 @@
+from __future__ import annotations
+
+"""
+zmap-tools — Python API for the Zebrafish Multi-Atlas Project (ZMAP).
+
+A curated single-cell RNA-seq reference atlas for zebrafish development,
+with tools for reference loading, label transfer, and visualization.
+
+Submodules
+----------
+reference (aliased as ref)
+    Download, cache, and load ZMAP reference H5ADs and consensus markers.
+predict
+    kNN-based label transfer from the ZMAP reference to query datasets.
+dotplot
+    Publication-ready dotplot visualization of gene expression patterns.
+"""
+
+from importlib.metadata import PackageNotFoundError, version as _version
+
+# Package version (from the PyPI/dist name)
+try:
+    __version__ = _version("zmap-tools")
+except PackageNotFoundError:
+    __version__ = "0.0.0"
+
+# Import subpackages as attributes 
+from . import reference    # zmap.reference
+from . import dotplot      # zmap.dotplot
+from . import predict      # zmap.predict
+
+# Scanpy-style aliases: 
+ref = reference
+
+# Top-level convenience APIs for biological module / baseline2 workflows.
+apply_encoder = predict.apply_encoder
+zmap_projection = predict.zmap_projection
+predict_pseudo_tissue = predict.predict_pseudo_tissue
+knn_config = predict.knn_config
+time_prediction_KNN = predict.time_prediction_KNN
+time_prediction_MLP = predict.time_prediction_MLP
+predict_labels_MLP = predict.predict_labels_MLP
+predict_labels_mlp = predict.predict_labels_mlp
+predict_labels_baseline2 = predict.predict_labels_baseline2
+predict_labels_baseline_tissue = predict.predict_labels_baseline_tissue
+baseline2 = predict.baseline2
+baseline_tissue = predict.baseline_tissue
+plot_conf_entropy_margin = predict.plot_conf_entropy_margin
+plot_pseudo_tissue_conf_entropy_margin = predict.plot_pseudo_tissue_conf_entropy_margin
+plot_confusion_matrix = predict.plot_confusion_matrix
+report_leiden_celltype_topk = predict.report_leiden_celltype_topk
+plot_leiden_celltype = predict.plot_leiden_celltype
+plot_projection_celltype_overlay = predict.plot_projection_celltype_overlay
+
+__all__ = ["__version__", "ref", "dotplot", "predict"]
+
+__all__ += [
+    "apply_encoder",
+    "zmap_projection",
+    "predict_pseudo_tissue",
+    "knn_config",
+    "time_prediction_KNN",
+    "time_prediction_MLP",
+    "predict_labels_MLP",
+    "predict_labels_mlp",
+    "predict_labels_baseline2",
+    "predict_labels_baseline_tissue",
+    "baseline2",
+    "baseline_tissue",
+    "plot_conf_entropy_margin",
+    "plot_pseudo_tissue_conf_entropy_margin",
+    "plot_confusion_matrix",
+    "report_leiden_celltype_topk",
+    "plot_leiden_celltype",
+    "plot_projection_celltype_overlay",
+]

{zmap_tools-0.2.2.2 → zmap_tools-0.2.2.3}/src/zmap/predict/__init__.py

@@ -33,6 +33,31 @@ from .predict import (
     validate_markers,
     plot_marker_comparison,
 )
+from .encoder_pipeline import (
+    apply_encoder,
+    zmap_projection,
+    predict_pseudo_tissue,
+    knn_config,
+    build_module_faiss,
+    build_module_concat_reference,
+    apply_module_concat_query,
+    time_prediction_KNN,
+    plot_conf_entropy_margin,
+    plot_pseudo_tissue_conf_entropy_margin,
+    plot_confusion_matrix,
+    report_leiden_celltype_topk,
+    plot_leiden_celltype,
+    plot_projection_celltype_overlay,
+)
+from .baseline2 import (
+    predict_labels_MLP,
+    predict_labels_mlp,
+    predict_labels_baseline2,
+    predict_labels_baseline_tissue,
+    time_prediction_MLP,
+    baseline2,
+    baseline_tissue,
+)
 
 __all__ = [
     "preprocess_adata_query",
@@ -55,3 +80,27 @@ __all__ = [
     "validate_markers",
     "plot_marker_comparison",
 ]
+
+__all__ += [
+    "apply_encoder",
+    "zmap_projection",
+    "predict_pseudo_tissue",
+    "knn_config",
+    "build_module_faiss",
+    "build_module_concat_reference",
+    "apply_module_concat_query",
+    "time_prediction_KNN",
+    "plot_conf_entropy_margin",
+    "plot_pseudo_tissue_conf_entropy_margin",
+    "plot_confusion_matrix",
+    "report_leiden_celltype_topk",
+    "plot_leiden_celltype",
+    "plot_projection_celltype_overlay",
+    "predict_labels_MLP",
+    "predict_labels_mlp",
+    "predict_labels_baseline2",
+    "predict_labels_baseline_tissue",
+    "time_prediction_MLP",
+    "baseline2",
+    "baseline_tissue",
+]