MLArray not support slice subscript

[annidy]

X_train_data = [[10, 100],
                [20, 150],
                [30, 120]]

X_train = MLArray(X_train_data)
print(X_train[:, 0])

this will output:

MLArray(shape=(2,),
data=[Value(data=10, grad=0), Value(data=100, grad=0)])

but in numpy, output will be:

array([10, 20, 30])

[rodmarkun]

Solved now in #13

The new implementation distinguishes between slices and integer indices:

if isinstance(curr_index, slice):
    # Apply slice to current dimension, then recurse on each element
    sliced_data = data[curr_index]
    if len(remaining_index) == 0:
        return sliced_data
    # Apply remaining indices to EACH element in the sliced result
    return [get_item_recursive(item, remaining_index) for item in sliced_data]
else:
    # Integer index: select that element and continue
    ...

Now with [:, 0]:
slice(None) selects all rows: [[10, 100], [20, 150], [30, 120]]
For each row, apply the remaining index (0,) → extracts element 0 from each
Result: [10, 20, 30]

Thanks! We can close the issue now