Pass missing stream arg in array.flatten - Githubissues

ml-explore / mlx

MLX: An array framework for Apple silicon

https://ml-explore.github.io/mlx/

MIT License

15.01k stars 856 forks source link

Pass missing stream arg in array.flatten #1111

Closed zcbenz closed 2 weeks ago

zcbenz commented 2 weeks ago

Proposed changes

The s arg is missing in the array.flatten implementation.

Checklist

Put an x in the boxes that apply.

[x] I have read the CONTRIBUTING document
[x] I have run pre-commit run --all-files to format my code / installed pre-commit prior to committing changes
[ ] I have added tests that prove my fix is effective or that my feature works
[ ] I have updated the necessary documentation (if needed)