Skip to content

Commit e925559

Browse files
committed
fix duplicate keys under batch mode
1 parent d4bb877 commit e925559

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

seqchromloader/writer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -192,6 +192,7 @@ def dump_data_webdataset_worker(coords,
192192
counter = 0
193193

194194
if batch_size is not None:
195+
feature_dict["__key__"] = ','.join(keys)
195196
feature_dict["seq.npy"] = np.array(seq_arr)
196197
feature_dict["chrom.npy"] = np.array(chrom_arr)
197198
feature_dict["target.npy"] = np.array(target_arr)

0 commit comments

Comments
 (0)