From 82d4d39225ec9bca2e83c21177c42f11013f9d86 Mon Sep 17 00:00:00 2001
From: Ori Kronfeld <ori.kronfeld@weizmann.ac.il>
Date: Tue, 31 Dec 2024 11:42:02 +0200
Subject: [PATCH] Update pyproject.toml

---
 pyproject.toml | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 7f573554a9..07600c3659 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -40,6 +40,7 @@ dependencies = [
     "lightning>=2.0",
     "ml-collections>=0.1.1",
     "mudata>=0.1.2",
+    "numba>=0.60.0"
     "numpy",
     "numpyro>=0.12.1",
     "pandas",
@@ -85,7 +86,7 @@ autotune = ["hyperopt>=0.2", "ray[tune]>=2.5.0"]
 # scvi.hub.HubModel.pull_from_s3
 aws = ["boto3"]
 # scvi.data.cellxgene
-census = ["cellxgene-census"]
+census = ["cellxgene-census","numpy<2.0"]
 # scvi.hub dependencies
 hub = ["huggingface_hub", "igraph", "leidenalg","dvc[s3]"]
 # scvi.data.add_dna_sequence
@@ -93,10 +94,10 @@ regseq = ["biopython>=1.81", "genomepy"]
 # scvi.criticism and read 10x
 scanpy = ["scanpy>=1.10","scikit-misc"]
 # for convinient files sharing
-file_share = ["pooch"]
+pooch = ["pooch"]
 
 optional = [
-    "scvi-tools[autotune,aws,hub,file_share,regseq,scanpy]"
+    "scvi-tools[autotune,aws,hub,pooch,regseq,scanpy]"
 ]
 tutorials = [
     "cell2location",