From a03c94a2063d146ad278e4d681fbce4f7f1afc9c Mon Sep 17 00:00:00 2001 From: Adil Date: Mon, 13 Apr 2026 22:37:47 +0530 Subject: [PATCH] fix: expose cohort size parameters in diversity_stats --- malariagen_data/anopheles.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/malariagen_data/anopheles.py b/malariagen_data/anopheles.py index 4df1d9d9c..ff2cea715 100644 --- a/malariagen_data/anopheles.py +++ b/malariagen_data/anopheles.py @@ -546,6 +546,8 @@ def diversity_stats( cohorts: base_params.cohorts, cohort_size: base_params.cohort_size, region: base_params.regions, + min_cohort_size: Optional[base_params.min_cohort_size] = None, + max_cohort_size: Optional[base_params.max_cohort_size] = None, site_mask: Optional[base_params.site_mask] = base_params.DEFAULT, site_class: Optional[base_params.site_class] = None, sample_query: Optional[base_params.sample_query] = None, @@ -564,7 +566,7 @@ def diversity_stats( sample_query=sample_query, sample_query_options=sample_query_options, cohort_size=cohort_size, - min_cohort_size=None, + min_cohort_size=min_cohort_size, ) # Compute diversity stats for cohorts. @@ -574,6 +576,8 @@ def diversity_stats( cohort=(cohort_label, cohort_query), cohort_size=cohort_size, region=region, + min_cohort_size=min_cohort_size, + max_cohort_size=max_cohort_size, site_mask=site_mask, site_class=site_class, sample_sets=sample_sets,