

def configObsIDsForEachBatch(self):
    ' Assign each observation in dataset to a batch by random permutation\n\n        Returns\n        --------\n        obsIDByBatch : list of length self.nBatch,\n                       where obsIDByBatch[bID] : list of all obsIDs in batch bID \n    '
    PRNG = np.random.RandomState(self.dataorderseed)
    docIDs = PRNG.permutation(self.nDocTotal).tolist()
    docIDs = [(x + 1) for x in docIDs]
    docIDByBatch = dict()
    for batchID in range(self.nBatch):
        docIDByBatch[batchID] = docIDs[:self.nObsBatch]
        del docIDs[:self.nObsBatch]
    return docIDByBatch

dataset/ETHPy150Open daeilkim/refinery/refinery/bnpy/bnpy-dev/bnpy/data/AdmixMinibatchIteratorDB.py AdmixMinibatchIteratorDB.configObsIDsForEachBatch/original