1
0
Fork 0

bug fix in covariate shift protocol

This commit is contained in:
Alejandro Moreo Fernandez 2022-06-17 12:51:52 +02:00
parent c0c37f0a17
commit cf0bd14cf1
1 changed files with 1 additions and 1 deletions

View File

@ -341,7 +341,7 @@ class CovariateShiftPP(AbstractStochasticSeededProtocol):
indexesA, indexesB = indexes indexesA, indexesB = indexes
sampleA = self.A.sampling_from_index(indexesA) sampleA = self.A.sampling_from_index(indexesA)
sampleB = self.B.sampling_from_index(indexesB) sampleB = self.B.sampling_from_index(indexesB)
return self.collator(sampleA+sampleB) return sampleA+sampleB
def total(self): def total(self):
return self.repeats * len(self.mixture_points) return self.repeats * len(self.mixture_points)