TYP: type annotations (#135)

* TYP: type annotations

* Python 3.9 fixes

* self-review

* code review

* Update array_api_strict/_array_object.py

Co-authored-by: Joren Hammudoglu <[email protected]>

* Apply suggestions from code review

Co-authored-by: Joren Hammudoglu <[email protected]>

* fix

* code review

* fixes

* normalize order

* Fancy indexing in `__getitem__` signature

* verbose Python scalar types

---------

Co-authored-by: Joren Hammudoglu <[email protected]>

Author: crusaderky

array_api_strict/__init__.py

@@ -16,6 +16,8 @@
 
 """
 
+from types import ModuleType
+
 __all__ = []
 
 # Warning: __array_api_version__ could change globally with
@@ -325,12 +327,16 @@
     ArrayAPIStrictFlags,
 )
 
-__all__ += ['set_array_api_strict_flags', 'get_array_api_strict_flags', 'reset_array_api_strict_flags', 'ArrayAPIStrictFlags']
+__all__ += [
+    'set_array_api_strict_flags',
+    'get_array_api_strict_flags',
+    'reset_array_api_strict_flags',
+    'ArrayAPIStrictFlags',
+    '__version__',
+]
 
 try:
-    from . import _version
-    __version__ = _version.__version__
-    del _version
+    from ._version import __version__  # type: ignore[import-not-found,unused-ignore]
 except ImportError:
     __version__ = "unknown"
 
@@ -340,7 +346,7 @@
 # use __getattr__. Note that linalg and fft are dynamically added and removed
 # from __all__ in set_array_api_strict_flags.
 
-def __getattr__(name):
+def __getattr__(name: str) -> ModuleType:
     if name in ['linalg', 'fft']:
         if name in get_array_api_strict_flags()['enabled_extensions']:
             if name == 'linalg':