Allow Image Source Method with random displacement in 2D (#315)

LCAV · May 25, 2023 · 7c8c0e3 · 7c8c0e3
1 parent 0ccf8ae
commit 7c8c0e3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyroomacoustics/room.py b/pyroomacoustics/room.py
@@ -2113,7 +2113,7 @@ def image_source_model(self):
                     max_disp = self.max_rand_disp
 
                     # add a random displacement to each cartesian coordinate
-                    disp = np.random.uniform(-max_disp, max_disp, size=(3, n_images))
+                    disp = np.random.uniform(-max_disp, max_disp, size=(self.dim, n_images))
                     source.images += disp
 
                 self.visibility.append(self.room_engine.visible_mics.copy())