From 65f9927abb68f30a79f0c7b76fbef515bdebf666 Mon Sep 17 00:00:00 2001 From: Tom White Date: Mon, 10 Feb 2025 10:44:35 +0000 Subject: [PATCH] Fix Cubed tests by chunking appropriately --- sgkit/tests/test_pca.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/sgkit/tests/test_pca.py b/sgkit/tests/test_pca.py index c132d0c7..1a8aca73 100644 --- a/sgkit/tests/test_pca.py +++ b/sgkit/tests/test_pca.py @@ -36,7 +36,7 @@ def simulate_dataset( n_variant: int = 100, n_sample: int = 50, n_cohort: Optional[int] = None, - chunks: Any = (None, None), + chunks: Any = (-1, -1), ) -> Dataset: """Simulate dataset with optional population structure""" ds = simulate_genotype_call_dataset(n_variant, n_sample, seed=0) @@ -48,6 +48,7 @@ def simulate_dataset( ac, dims=("variants", "samples") ) else: + ds["call_genotype_mask"] = ds["call_genotype_mask"].chunk(chunks + (1,)) ds = count_call_alternate_alleles(ds) ds["call_alternate_allele_count"] = ds["call_alternate_allele_count"].chunk(chunks) return ds