Python 3.9 fixes

Author: crusaderky

array_api_strict/_array_object.py

@@ -18,7 +18,7 @@
 import operator
 from collections.abc import Iterator
 from enum import IntEnum
-from types import EllipsisType, ModuleType
+from types import ModuleType
 from typing import Any, Literal, SupportsIndex
 
 import numpy as np
@@ -42,6 +42,10 @@
 from ._flags import get_array_api_strict_flags, set_array_api_strict_flags
 from ._typing import PyCapsule
 
+try:
+    from types import EllipsisType  # Python >=3.10
+except ImportError:
+    EllipsisType = type(Ellipsis)
 
 class Device:
     _device: str
@@ -345,7 +349,7 @@ def _validate_index(
             | tuple[int | slice | EllipsisType | None, ...]
             | Array
         ),
-        op: Literal["getitem", "setitem"],
+        op: Literal["getitem", "setitem"] = "getitem",
     ) -> None:
         """
         Validate an index according to the array API.

array_api_strict/_dtypes.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import warnings
 from typing import Any
 

array_api_strict/_helpers.py

@@ -1,5 +1,6 @@
 """Private helper routines.
 """
+from __future__ import annotations
 
 from ._array_object import Array
 from ._dtypes import _dtype_categories