diff --git a/examples/photon_histogram.py b/examples/photon_histogram.py index 3a2400597..9e0d281d4 100644 --- a/examples/photon_histogram.py +++ b/examples/photon_histogram.py @@ -31,5 +31,4 @@ class PhotonHistogram(AutoContext): n = self.nbins-1 hist[n] += 1 - for i in range(self.nbins): - print(i, hist[i]) + print(hist) diff --git a/examples/transport.py b/examples/transport.py index 9deff73dd..5f1d92bff 100644 --- a/examples/transport.py +++ b/examples/transport.py @@ -94,16 +94,13 @@ class Transport(AutoContext): @kernel def repeat(self): - hist = [0 for _ in range(self.nbins)] + self.histogram = [0 for _ in range(self.nbins)] for i in range(self.repeats): n = self.one() if n >= self.nbins: n = self.nbins-1 - hist[n] += 1 - - for i in range(self.nbins): - self.histogram.append(hist[i]) + self.histogram[n] += 1 def scan(self, stops): for s in stops: