A little more cleanup

Author: garrettwrong

src/aspire/classification/averager2d.py

@@ -17,6 +17,10 @@
 def commute_shift_rot(shifts, rots):
     """
     Rotate `shifts` points by `rots` ccw radians.
+
+    :param shifts: Array of shift points shaped (..., 2)
+    :param rots: Array of rotations (radians)
+    :returns: Array of rotated shift points shaped (..., 2)
     """
     sx = shifts[:, 0]
     sy = shifts[:, 1]
@@ -778,12 +782,6 @@ def __init__(
 
         self.n_angles = n_angles
 
-        # # XXX Will use polar for rotate
-        # if not hasattr(self.alignment_basis, "rotate"):
-        #     raise RuntimeError(
-        #         f"{self.__class__.__name__}'s alignment_basis {self.alignment_basis} must provide a `rotate` method."
-        #     )
-
         self.radius = radius if radius is not None else src.L // 16
 
         if self.radius != 0:

src/aspire/image/image.py

@@ -668,7 +668,7 @@ def _im_translate(self, shifts):
         # Note original stack shape and flatten stack
         stack_shape = self.stack_shape
         if self.n_images == 1 and n_shifts > 1:
-            # XXX special case, cleanup broadcasting later
+            # Handle the shift broadcast special case
             stack_shape = n_shifts
         im = self.stack_reshape(-1)._data