def configObsIDsForEachBatch(self):
    PRNG = np.random.RandomState(self.dataorderseed)
    docIDs = PRNG.permutation(self.nDocTotal).tolist()
    # need to add 1 to list of document indices since sql indexes rows starting with 1
    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
Assign each observation in dataset to a batch by random permutation
dataset/ETHPy150Open daeilkim/refinery/refinery/bnpy/bnpy-dev/bnpy/data/AdmixMinibatchIteratorDB.py/AdmixMinibatchIteratorDB.configObsIDsForEachBatch