Use 'from __future__ import annotations' in the function stubs

Otherwise the annotation for -inf will fail, since inf is stubbed as None. We
also don't need to evaluate the other annotations either, even though they
shouldn't fail.

Author: asmeurer

array_api_tests/function_stubs/array_object.py

@@ -14,6 +14,8 @@
 2. There is no real way to test that anyway.
 """
 
+from __future__ import annotations
+
 from ._types import array
 
 def __abs__(x: array) -> array:

array_api_tests/function_stubs/constants.py

@@ -14,6 +14,8 @@
 2. There is no real way to test that anyway.
 """
 
+from __future__ import annotations
+
 
 e = None
 

array_api_tests/function_stubs/creation_functions.py

@@ -14,6 +14,8 @@
 2. There is no real way to test that anyway.
 """
 
+from __future__ import annotations
+
 from ._types import Optional, Tuple, Union, array, device, dtype
 
 def arange(start: Union[int, float], *, stop: Optional[Union[int, float]] = None, step: Union[int, float] = 1, dtype: Optional[dtype] = None, device: Optional[device] = None) -> array:

array_api_tests/function_stubs/elementwise_functions.py

@@ -14,6 +14,8 @@
 2. There is no real way to test that anyway.
 """
 
+from __future__ import annotations
+
 from ._types import array
 
 def abs(x: array) -> array:

array_api_tests/function_stubs/linear_algebra_functions.py

@@ -14,6 +14,8 @@
 2. There is no real way to test that anyway.
 """
 
+from __future__ import annotations
+
 from ._types import Literal, Optional, Tuple, Union, array
 from .constants import inf
 

array_api_tests/function_stubs/manipulation_functions.py

@@ -14,6 +14,8 @@
 2. There is no real way to test that anyway.
 """
 
+from __future__ import annotations
+
 from ._types import Optional, Tuple, Union, array
 
 def concat(arrays: Tuple[array], *, axis: Optional[int] = 0) -> array:

array_api_tests/function_stubs/searching_functions.py

@@ -14,6 +14,8 @@
 2. There is no real way to test that anyway.
 """
 
+from __future__ import annotations
+
 from ._types import Tuple, array
 
 def argmax(x: array, *, axis: int = None, keepdims: bool = False) -> array:

array_api_tests/function_stubs/set_functions.py

@@ -14,6 +14,8 @@
 2. There is no real way to test that anyway.
 """
 
+from __future__ import annotations
+
 from ._types import Tuple, Union, array
 
 def unique(x: array, *, return_counts: bool = False, return_index: bool = False, return_inverse: bool = False, sorted: bool = True) -> Union[array, Tuple[array, ...]]:

array_api_tests/function_stubs/sorting_functions.py

@@ -14,6 +14,8 @@
 2. There is no real way to test that anyway.
 """
 
+from __future__ import annotations
+
 from ._types import array
 
 def argsort(x: array, *, axis: int = -1, descending: bool = False, stable: bool = True) -> array:

array_api_tests/function_stubs/statistical_functions.py

@@ -14,6 +14,8 @@
 2. There is no real way to test that anyway.
 """
 
+from __future__ import annotations
+
 from ._types import Optional, Tuple, Union, array
 
 def max(x: array, *, axis: Optional[Union[int, Tuple[int, ...]]] = None, keepdims: bool = False) -> array:

array_api_tests/function_stubs/utility_functions.py

@@ -14,6 +14,8 @@
 2. There is no real way to test that anyway.
 """
 
+from __future__ import annotations
+
 from ._types import Optional, Tuple, Union, array
 
 def all(x: array, *, axis: Optional[Union[int, Tuple[int, ...]]] = None, keepdims: bool = False) -> array:

generate_stubs.py

@@ -43,6 +43,8 @@
 2. There is no real way to test that anyway.
 """
 
+from __future__ import annotations
+
 from ._types import *
 from .constants import *
 '''