minor fix due to numpy bahavior diff across versions

Author: yztxwd

tpcav/concepts.py

@@ -62,7 +62,6 @@ def _construct_motif_concept_dataloader_from_control(
         wds.RandomMix(datasets),
         batch_size=batch_size,
         num_workers=num_workers,
-        pin_memory=True,
         drop_last=False,
     )
     return mixed_dl

tpcav/utils.py

@@ -192,9 +192,7 @@ def __iter__(self):
             if worker_info is None:
                 chunk = self.seq_df
             else:
-                chunk = np.array_split(self.seq_df, worker_info.num_workers)[
-                    worker_info.id
-                ]
+                chunk = [self.seq_df.iloc[idx] for idx in np.array_split(np.arange(len(self.seq_df)), worker_info.num_workers)][worker_info.id]
             yield from iterate_seq_df_chunk(
                 chunk,
                 genome=self.genome,