[Core][NDArray][ComplexNDArray] Improvements to __getitem__ method of Arrays

numojo/__init__.mojo

@@ -19,33 +19,44 @@ from numojo.core.complex.complex_ndarray import ComplexNDArray
 from numojo.core.complex.complex_dtype import (
     ComplexDType,
     ci8,
-    ci16,
-    ci32,
     ci64,
-    cisize,
-    cintp,
+    ci128,
+    ci256,
+    cint,
     cu8,
     cu16,
     cu32,
     cu64,
+    cu128,
+    cu256,
+    cuint,
+    cbf16,
     cf16,
     cf32,
     cf64,
     cboolean,
+    cinvalid,
 )
 from numojo.core.datatypes import (
     i8,
     i16,
     i32,
     i64,
-    isize,
+    i128,
+    i256,
+    int,
     u8,
     u16,
     u32,
     u64,
+    u128,
+    u256,
+    uint,
+    bf16,
     f16,
     f32,
     f64,
+    boolean,
 )
 from numojo.core.error import (
     ShapeError,

numojo/core/__init__.mojo

@@ -13,33 +13,43 @@ from .complex import (
     ComplexNDArray,
     ComplexDType,
     ci8,
-    ci16,
-    ci32,
     ci64,
-    cisize,
-    cintp,
+    ci128,
+    ci256,
+    cint,
     cu8,
     cu16,
     cu32,
     cu64,
+    cu128,
+    cu256,
+    cuint,
+    cbf16,
     cf16,
     cf32,
     cf64,
     cboolean,
+    cinvalid,
 )
 
 from .datatypes import (
     i8,
-    i16,
-    i32,
     i64,
+    i128,
+    i256,
+    int,
     u8,
     u16,
     u32,
     u64,
+    u128,
+    u256,
+    uint,
+    bf16,
     f16,
     f32,
     f64,
+    boolean,
 )
 
 from .error import (

numojo/core/complex/__init__.mojo

@@ -3,17 +3,21 @@ from .complex_ndarray import ComplexNDArray
 from .complex_dtype import (
     ComplexDType,
     ci8,
-    ci16,
-    ci32,
     ci64,
-    cisize,
-    cintp,
+    ci128,
+    ci256,
+    cint,
     cu8,
     cu16,
     cu32,
     cu64,
+    cu128,
+    cu256,
+    cuint,
+    cbf16,
     cf16,
     cf32,
     cf64,
     cboolean,
+    cinvalid,
 )