Skip to content

Commit c5de286

Browse files
committed
change df split way to avoid numpy behavior diff
1 parent ad6b754 commit c5de286

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

seqchromloader/writer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ def dump_data_webdataset(coords, genome_fasta, bigwig_filelist,
9999

100100
# split coordinates and assign chunks to workers
101101
num_chunks = math.ceil(len(coords) / samples_per_tar)
102-
chunks = np.array_split(coords, num_chunks)
102+
chunks = [coords.iloc[idx] for idx in np.array_split(np.arange(len(coords)), num_chunks)]
103103

104104
# freeze the common parameters
105105
## create a scaler to get statistics for normalizing chromatin marks input

0 commit comments

Comments
 (0)