FIX: Wrap torch.argsort to set stable=True by default

array_api_compat/torch/_aliases.py

@@ -241,6 +241,20 @@ def sort(
 ) -> Array:
     return torch.sort(x, dim=axis, descending=descending, stable=stable, **kwargs).values
 
+
+# Wrap torch.argsort to set stable=True by default
+def argsort(
+    x: Array,
+    /,
+    *,
+    axis: int = -1,
+    descending: bool = False,
+    stable: bool = True,
+    **kwargs: object,
+) -> Array:
+    return torch.argsort(x, dim=axis, descending=descending, stable=stable, **kwargs)
+
+
 def _normalize_axes(axis, ndim):
     axes = []
     if ndim == 0 and axis:

array_api_compat/torch/fft.py

@@ -3,7 +3,7 @@
 from collections.abc import Sequence
 from typing import Literal
 
-import torch
+import torch  # noqa: F401
 import torch.fft
 
 from ._typing import Array