lint

Author: crusaderky

array_api_compat/numpy/fft.py

@@ -20,7 +20,7 @@
 ifftshift = get_xp(np)(_fft.ifftshift)
 
 
-__all__ = _fft.__all__ + ["rfft2", "irfft2", "fft2", "ifft2"]
+__all__ = _fft.__all__ + ["fft2", "ifft2", "irfft2", "rfft2"]
 
 def __dir__() -> list[str]:
     return __all__

tests/test_all.py

@@ -312,10 +312,7 @@ def test_compat_spurious_names(library, module):
     compat_mod = getattr(compat_xp, module) if module else compat_xp
     aapi_names = set(NAMES[module])
     compat_spurious_names = (
-        set(dir(compat_mod))
-        - set(dir(bare_mod))
-        - aapi_names 
-        - {"__all__"}
+        set(dir(compat_mod)) - set(dir(bare_mod)) - aapi_names - {"__all__"}
     )
     # Quietly ignore *Result dataclasses
     compat_spurious_names = {