Merge pull request #3 from SongY123/main

fix bugs when indice is numpy object
BUAA-BDA · Jan 13, 2025 · d419b0e · d419b0e
2 parents 44e7a1a + 909ec40
commit d419b0e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/openhufu/dataset/splitters/generic/iid_splitter.py b/openhufu/dataset/splitters/generic/iid_splitter.py
@@ -17,5 +17,5 @@ def __call__(self, dataset):
         indices = np.arange(len(dataset))
         np.random.shuffle(indices)
         idx_slices = np.array_split(indices, self.n_clients)
-        subsets = [self.subset(dataset, idxs) for idxs in idx_slices]
+        subsets = [self.subset(dataset, [int(idx) for idx in idxs]) for idxs in idx_slices]
         return subsets
diff --git a/openhufu/dataset/splitters/generic/lda_splitter.py b/openhufu/dataset/splitters/generic/lda_splitter.py
@@ -51,5 +51,5 @@ def __call__(self, dataset, get_labels_fn, alpha=0.5, **kwargs):
             for client_id, split in enumerate(splits):
                 client_indices[client_id].extend(split)
 
-        subsets = [self.subset(dataset, idxs) for idxs in client_indices]
+        subsets = [self.subset(dataset, [int(idx) for idx in idxs]) for idxs in client_indices]
         return subsets