Fix random generation of single floats

QuLogic · QuLogic · commit f8f37c30090a · 2022-12-20T03:18:36.000-05:00
Passing 1 means we get a 1D array with a single element, but passing
nothing means a 0D array, aka a single float.
diff --git a/examples/animation/bayes_update.py b/examples/animation/bayes_update.py
@@ -47,7 +47,7 @@ def __call__(self, i):
             return self.line,
 
         # Choose success based on exceed a threshold with a uniform pick
-        if np.random.rand(1,) < self.prob:
+        if np.random.rand() < self.prob:
             self.success += 1
         y = beta_pdf(self.x, self.success + 1, (i - self.success) + 1)
         self.line.set_data(self.x, y)
diff --git a/examples/animation/strip_chart.py b/examples/animation/strip_chart.py
@@ -37,19 +37,19 @@ def update(self, y):
         t = self.tdata[0] + len(self.tdata) * self.dt
 
         self.tdata.append(t)
-        self.ydata.append(float(y))
+        self.ydata.append(y)
         self.line.set_data(self.tdata, self.ydata)
         return self.line,
 
 
 def emitter(p=0.1):
     """Return a random value in [0, 1) with probability p, else 0."""
     while True:
-        v = np.random.rand(1)
+        v = np.random.rand()
         if v > p:
             yield 0.
         else:
-            yield np.random.rand(1)
+            yield np.random.rand()
 
 
 # Fixing random state for reproducibility
