fix: typo

js2264 · Oct 19, 2024 · 609484c · 609484c
1 parent 27dc618
commit 609484c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)