diff --git a/pyproject.toml b/pyproject.toml index 3f53bfb1..5986cca3 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "arc-state" -version = "0.10.4" +version = "0.10.5" description = "State is a machine learning model that predicts cellular perturbation response across diverse contexts." readme = "README.md" authors = [ @@ -27,7 +27,7 @@ dependencies = [ "geomloss>=0.2.6", "transformers>=4.52.3", "peft>=0.11.0", - "cell-eval>=0.6.8", + "cell-eval>=0.7.0", "ipykernel>=6.30.1", "scipy>=1.15.0", ] diff --git a/src/state/_cli/_tx/_predict.py b/src/state/_cli/_tx/_predict.py index 7c88929e..41c7e67a 100644 --- a/src/state/_cli/_tx/_predict.py +++ b/src/state/_cli/_tx/_predict.py @@ -578,7 +578,6 @@ def ensure_list(values, batch_size: int): outdir=results_dir, prefix=ct, pdex_kwargs=pdex_kwargs, - batch_size=2048, ) evaluator.compute( profile=args.profile, @@ -812,7 +811,6 @@ def ensure_list(values, batch_size: int): outdir=results_dir, prefix=ct, pdex_kwargs=pdex_kwargs, - batch_size=2048, ) evaluator.compute(