Skip to content

Commit 1d8f16f

Browse files
authored
Merge pull request #154 from zudi-lin/volume_ds
fix volume dataset
2 parents 6fc4435 + a551dd9 commit 1d8f16f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

connectomics/data/dataset/dataset_volume.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,7 @@ def __init__(self,
153153
self.valid_ratio = valid_ratio
154154
# precompute valid region
155155
# can be memory intensive
156-
self.valid_pos = [None] * len(self.valid_mask)
156+
self.valid_pos = [None] * len(self.valid_mask) if self.valid_mask is not None else [None] * len(self.volume)
157157
"""
158158
if self.valid_mask is not None:
159159
for i, x in enumerate(self.valid_mask):

0 commit comments

Comments
 (0)