multi thread debug

jkobject · Jun 10, 2024 · f4e54a9 · f4e54a9
1 parent 5219e03
commit f4e54a9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/scdataloader/datamodule.py b/scdataloader/datamodule.py
@@ -7,7 +7,7 @@
     SubsetRandomSampler,
     SequentialSampler,
     RandomSampler,
-)
+) 
 import torch
 from torch.utils.data import DataLoader, Sampler
 import lightning as L
@@ -29,7 +29,7 @@ def __init__(
         train_oversampling_per_epoch: float = 0.1,
         validation_split: float = 0.2,
         test_split: float = 0,
-        gene_embeddings: str = "",
+        gene_embeddings: str = "", 
         use_default_col: bool = True,
         gene_position_tolerance: int = 10_000,
         # this is for the mappedCollection

diff --git a/scdataloader/mapped.py b/scdataloader/mapped.py
@@ -245,6 +245,8 @@ def _make_encoders(self, encode_labels: list):
             if unknown_label is not None and unknown_label in cats:
                 cats.remove(unknown_label)
                 encoder[unknown_label] = -1
+            cats = list(cats)
+            cats.sort()
             encoder.update({cat: i for i, cat in enumerate(cats)})
             self.encoders[label] = encoder