convert arange to a protocol

Signed-off-by: nstarman <[email protected]>

Author: nstarman

Diff for: src/array_api_stubs/_draft/_namespace.py

@@ -2,7 +2,10 @@
 
 from typing import Protocol
 
+from .creation_functions import arange as ArangeCallable
+
 
 class ArrayAPINamespace(Protocol):
     """Protocol for the array API namespace itself."""
 
+    arange: ArangeCallable

Diff for: src/array_api_stubs/_draft/creation_functions.py

@@ -1,3 +1,4 @@
+from typing import Protocol
 from ._types import (
     List,
     NestedSequence,
@@ -11,15 +12,7 @@
 )
 
 
-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:
+class arange(Protocol):
     """
     Returns evenly spaced values within the half-open interval ``[start, stop)`` as a one-dimensional array.
 
@@ -46,6 +39,17 @@ def arange(
         a one-dimensional array containing evenly spaced values. The length of the output array must be ``ceil((stop-start)/step)`` if ``stop - start`` and ``step`` have the same sign, and length ``0`` otherwise.
     """
 
+    def __call__(
+        start: Union[int, float],
+        /,
+        stop: Optional[Union[int, float]] = None,
+        step: Union[int, float] = 1,
+        *,
+        dtype: Optional[dtype] = None,
+        device: Optional[device] = None,
+    ) -> array:
+        ...
+
 
 def asarray(
     obj: Union[