_:hammer_and_wrench: Refactor suggestion_

[mmcdermott]

          _:hammer_and_wrench: Refactor suggestion_

Use enumerate for Cleaner Looping

In _slice_single, replace manual index tracking with enumerate:

-    i = 0
-    for dim in sorted(squeeze_dims):
-        out = out.squeeze(dim - i)
-        i += 1
+    for i, dim in enumerate(sorted(squeeze_dims)):
+        out = out.squeeze(dim - i)

This change improves clarity and conciseness.

Committable suggestion was skipped due to low confidence.

🧰 Tools

🪛 Ruff

1261-1261: Use enumerate() for index variable i in for loop

(SIM113)

Originally posted by @coderabbitai[bot] in #35 (comment)