Clean up

Author: adonath

array_api_compat/__init__.py

@@ -17,7 +17,7 @@
 this implementation for the default when working with NumPy arrays.
 
 """
-__version__ = "1.4.1"
+__version__ = '1.4.1'
 
 from .common import (
  array_namespace,

array_api_compat/_internal.py

@@ -44,7 +44,7 @@ def wrapped_f(*args, **kwargs):
  return inner
 
 
-def get_all_public_members(module, filter_=None):
+def _get_all_public_members(module, filter_=None):
  """Get all public members of a module."""
  try:
  return getattr(module, '__all__')
@@ -54,4 +54,4 @@ def get_all_public_members(module, filter_=None):
  if filter_ is None:
  filter_ = lambda name: name.startswith('_') # noqa: E731
 
- return map(dir(module), filter_)
+ return map(filter_, dir(module))

array_api_compat/common/_helpers.py

@@ -7,8 +7,11 @@
 """
 from __future__ import annotations
 
-from typing import Optional, Union, Any
-from ._typing import Array, Device
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+ from typing import Optional, Union, Any
+ from ._typing import Array, Device 
 
 import sys
 import math