Merge pull request #29 from TomGeorge1234/1.x

Various merges made into 1.x
RatInABox-Lab · Mar 16, 2023 · 89242b9 · 89242b9
2 parents 0b51c34 + bd35431
commit 89242b9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ratinabox/Environment.py b/ratinabox/Environment.py
@@ -284,9 +284,9 @@ def sample_positions(self, n=10, method="uniform_jitter"):
         elif self.dimensionality == "2D":
 
             if method == "random":
-                positions = np.random.uniform(size=(n, 2))
-                positions[:, 0] *= self.extent[1] - self.extent[0]
-                positions[:, 1] *= self.extent[3] - self.extent[2]
+                positions = np.zeros((n, 2))
+                positions[:, 0] = np.random.uniform(self.extent[0], self.extent[1], size=n)
+                positions[:, 1] = np.random.uniform(self.extent[2], self.extent[3], size=n)
             elif method[:7] == "uniform":
                 ex = self.extent
                 area = (ex[1] - ex[0]) * (ex[3] - ex[2])