diff --git a/pyproject.toml b/pyproject.toml index e6eb7541..d0dd62e6 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "arc-state" -version = "0.9.32" +version = "0.9.33" description = "State is a machine learning model that predicts cellular perturbation response across diverse contexts." readme = "README.md" authors = [ diff --git a/src/state/_cli/_tx/_predict.py b/src/state/_cli/_tx/_predict.py index 6967978f..b0bbecd7 100644 --- a/src/state/_cli/_tx/_predict.py +++ b/src/state/_cli/_tx/_predict.py @@ -52,6 +52,13 @@ def add_arguments_predict(parser: ap.ArgumentParser): help="If set, evaluate the model on the training data rather than on the test data.", ) + parser.add_argument( + "--fdr-threshold", + type=float, + default=1e-3, + help="FDR threshold for DE significance in cell-eval metrics. Default is 1e-3.", + ) + def run_tx_predict(args: ap.ArgumentParser): import logging @@ -422,6 +429,7 @@ def load_config(cfg_path: str) -> dict: prefix=ct, pdex_kwargs=pdex_kwargs, batch_size=2048, + fdr_threshold=args.fdr_threshold, ) evaluator.compute(