fix: set dataloader kwargs default as {}

Author: yztxwd

seqchromloader/loader.py

@@ -29,7 +29,7 @@ class SeqChromLoader():
     def __init__(self, SeqChromDataset):
         self.SeqChromDataset = SeqChromDataset
 
-    def __call__(self, *args, dataloader_kws:dict=None, **kwargs):
+    def __call__(self, *args, dataloader_kws:dict={}, **kwargs):
         # default dataloader kws
         if dataloader_kws is not None:
             wif = dataloader_kws.pop("worker_init_fn", worker_init_fn)
@@ -268,4 +268,4 @@ def val_dataloader(self):
         return wds.WebLoader(self.val_loader, num_workers=self.num_workers, batch_size=self.batch_size_per_rank)
 
     def test_dataloader(self):
-        return wds.WebLoader(self.test_loader, num_workers=self.num_workers, batch_size=self.batch_size_per_rank)
+        return wds.WebLoader(self.test_loader, num_workers=self.num_workers, batch_size=self.batch_size_per_rank)