def set_migrations(self, data): print("Fitting migration distribution") r = [] for i in range(len(data)): num = data[i] for _ in range(num): r.append(i) dist = Distribution() dist.min_val = 0 dist.max_val = len(data) - 1 dist.Fit(r) self.migrations_distribution = dist