From 57e920e195f4ca06bf309948beca0f4a15a4cd44 Mon Sep 17 00:00:00 2001 From: Adam Pooley Date: Sat, 11 Jul 2026 01:17:45 -0400 Subject: [PATCH] Fixes set_velocities to accept ids as scalar or 1d array --- rps/robotarium_abc.py | 1 + 1 file changed, 1 insertion(+) diff --git a/rps/robotarium_abc.py b/rps/robotarium_abc.py index bfc7e13..9d4059c 100644 --- a/rps/robotarium_abc.py +++ b/rps/robotarium_abc.py @@ -126,6 +126,7 @@ def set_velocities(self, ids: NDArray[np.integer], velocities: NDArray[np.floati assert velocities.shape[1] <= self.number_of_robots, ( f"Column count of velocities ({velocities.shape[1]}) must be <= number_of_robots ({self.number_of_robots})" ) + ids = np.atleast_1d(ids) self._velocities[:, ids] = velocities def robot_colormap(self) -> plt.cm.ScalarMappable: