tiny debug

jkobject · Apr 9, 2024 · c8bf334 · c8bf334
1 parent 33184c5
commit c8bf334
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/scdataloader/collator.py b/scdataloader/collator.py
@@ -113,7 +113,7 @@ def __call__(self, batch):
             organism_id = elem[self.organism_name]
             if organism_id not in self.organism_ids:
                 continue
-            if "dataset" in elem:
+            if "_storage_idx" in elem:
                 dataset.append(elem["_storage_idx"])
             expr = np.array(elem["x"])
             total_count.append(expr.sum())

diff --git a/scdataloader/datamodule.py b/scdataloader/datamodule.py
@@ -154,7 +154,9 @@ def __repr__(self):
             f"\ttest_split={self.test_split},\n"
             f"\tn_samples={self.n_samples},\n"
             f"\tweight_scaler={self.weight_scaler},\n"
-            f"\train_oversampling_per_epoch={self.train_oversampling_per_epoch},\n"
+            f"\ttrain_oversampling_per_epoch={self.train_oversampling_per_epoch},\n"
+            f"\tassays_to_drop={self.assays_to_drop},\n"
+            f"\tnum_datasets={len(self.dataset.mapped_dataset.storages)},\n"
             f"\tlabel_to_weight={self.label_to_weight}\n"
             + (
                 "\twith train_dataset size of=("