a43bba5127d2539e2a3741e2f20247012f5dd746,pyprob/model.py,Model,_prior_traces,#Model#Any#Any#Any#,37

Before Change

    def _prior_traces(self, samples=10, trace_state=TraceState.RECORD, proposal_network=None, *args, **kwargs):
        generator = self._prior_trace_generator(trace_state, proposal_network, *args, **kwargs)
        return [next(generator) for i in range(samples)]

    def prior_sample(self, *args, **kwargs):
        generator = self._prior_sample_generator(*args, **kwargs)

After Change

    def _prior_traces(self, samples=10, trace_state=TraceState.RECORD, proposal_network=None, *args, **kwargs):
        generator = self._prior_trace_generator(trace_state, proposal_network, *args, **kwargs)
        ret = []
        time_start = time.time()
        for i in range(samples):
            print("                                                                \r{} | {} / {} | {:,} traces/s".format(util.progress_bar(i+1, samples), i+1, samples, int(i / (time.time() - time_start))), end="\r")
            ret.append(next(generator))
        return ret

Instances