remove arange

Author: AllanZyne

examples/transpose.py

@@ -50,7 +50,7 @@ def np_transpose(arr):
 
 
 def initialize(np, row, col, dtype):
-    arr = np.arange(row * col, dtype=dtype)
+    arr = np.arange(0, row * col, dtype=dtype)
     return np.reshape(arr, (row, col))
 
 

sharpy/__init__.py

@@ -96,17 +96,6 @@ def _validate_device(device):
         raise ValueError(f"Invalid device string: {device}")
 
 
-def arange(start, stop=None, step=1, dtype=int64, device="", team=1):
-    if stop is None:
-        stop = start
-        start = 0
-    return ndarray(
-        _csp.Creator.arange(
-            start, stop, step, dtype, _validate_device(device), team
-        )
-    )
-
-
 for func in api.api_categories["Creator"]:
     FUNC = func.upper()
     if func == "full":
@@ -125,6 +114,10 @@ def arange(start, stop=None, step=1, dtype=int64, device="", team=1):
         exec(
             f"{func} = lambda shape, dtype=float64, device='', team=1: ndarray(_csp.Creator.full(shape, 0, dtype, _validate_device(device), team))"
         )
+    elif func == "arange":
+        exec(
+            f"{func} = lambda start, end, step, dtype=int64, device='', team=1: ndarray(_csp.Creator.arange(start, end, step, dtype, _validate_device(device), team))"
+        )
     elif func == "linspace":
         exec(
             f"{func} = lambda start, end, step, endpoint, dtype=float64, device='', team=1: ndarray(_csp.Creator.linspace(start, end, step, endpoint, dtype, _validate_device(device), team))"