fe6d7f916a0e9d1a0f42626f3747ed14f9e3f684,pyprob/model.py,Model,trace_length_mean,#Model#Any#,205

Before Change

    def trace_length_mean(self, num_traces=1000, *args, **kwargs):
        traces = self._prior_traces(num_traces, trace_mode=TraceMode.RECORD, inference_network=None, *args, **kwargs)
        trace_length_dist = Empirical([trace.length for trace in traces])
        return trace_length_dist.mean

    def trace_length_stddev(self, num_traces=1000, *args, **kwargs):

After Change

    def trace_length_mean(self, num_traces=1000, *args, **kwargs):
        trace_lengths = self._prior_traces(num_traces, trace_mode=TraceMode.RECORD, inference_network=None, map_func=lambda trace: trace.length, *args, **kwargs)
        trace_length_dist = Empirical(trace_lengths)
        return trace_length_dist.mean

    def trace_length_stddev(self, num_traces=1000, *args, **kwargs):
        trace_lengths = self._prior_traces(num_traces, trace_mode=TraceMode.RECORD, inference_network=None, map_func=lambda trace: trace.length, *args, **kwargs)

Instances