Skip to content

Commit 9f1197e

Browse files
authored
fix channel checking for dataset-write (#208)
1 parent 889ac9e commit 9f1197e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

wkcuber/api/MagDataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,7 @@ def _assert_valid_num_channels(self, write_data_shape):
105105
), f"The number of channels of the dataset ({num_channels}) does not match the number of channels of the passed data (1)"
106106
else:
107107
assert (
108-
num_channels == 3
108+
num_channels == write_data_shape[0]
109109
), f"The number of channels of the dataset ({num_channels}) does not match the number of channels of the passed data ({write_data_shape[0]})"
110110

111111

0 commit comments

Comments
 (0)