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