From 609484c62d08975f8a3c30028d27b1ae6354710b Mon Sep 17 00:00:00 2001 From: js2264 Date: Sat, 19 Oct 2024 03:23:56 +0200 Subject: [PATCH] fix: typo --- tests/test_data.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_data.py b/tests/test_data.py index be5f704..4855fe9 100644 --- a/tests/test_data.py +++ b/tests/test_data.py @@ -42,7 +42,7 @@ def test_dataset(momics_path: str): assert n[0].shape == (10, 10, 1) assert n[1].shape == (10, 2, 1) - rg = MomicsDataset(mom, b, "nucleotide", "CH1", target_size=2, batch_size=10) + rg = MomicsDataset(mom, b, "nucleotide", "bw2", target_size=2, batch_size=10) n = next(iter(rg)) assert n[0].shape == (10, 10, 4) assert n[1].shape == (10, 2, 1)