Fix final ruff errors in array_api_compat/torch/__init__.py

Author: adonath

array_api_compat/torch/__init__.py

@@ -2,19 +2,28 @@
 import torch
 from torch import * # noqa: F401, F403
 
-for n in dir(torch):
+from .._internal import _get_all_public_members
+
+
+def filter_(name):
  if (
- n.startswith("_")
- or n.endswith("_")
- or "cuda" in n
- or "cpu" in n
- or "backward" in n
+ name.startswith("_")
+ or name.endswith("_")
+ or "cuda" in name
+ or "cpu" in name
+ or "backward" in name
  ):
- continue
- exec(n + " = torch." + n)
+ return False
+ return True
+
+
+_torch_all = _get_all_public_members(torch, filter_=filter_)
 
+for _name in _torch_all:
+ globals()[_name] = getattr(torch, _name)
 
-from ..common._helpers import (
+
+from ..common._helpers import ( # noqa: E402
  array_namespace,
  device,
  get_namespace,
@@ -24,7 +33,7 @@
 )
 
 # These imports may overwrite names from the import * above.
-from ._aliases import (
+from ._aliases import ( # noqa: E402
  add,
  all,
  any,
@@ -92,7 +101,11 @@
  zeros,
 )
 
-__all__ = [
+__all__ = []
+
+__all__ += _torch_all
+
+__all__ += [
  "is_array_api_obj",
  "array_namespace",
  "get_namespace",