zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-08 01:54:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: handle linspace function when num is 1 (#602)

Browse Source 
* fix: handle linspace function when num is 1

* add comment

* fix test case

* remove breakpoint
This commit is contained in:
Avikant Srivastava
2024-02-05 00:33:49 +05:30
committed by GitHub
parent 4fd2fb84a6
commit 11a9fd40f0
2 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
python/tests/test_ops.py
View File

@@ -1661,6 +1661,11 @@ class TestOps(mlx_tests.MLXTestCase):
expected = mx.array(np.linspace(0, 1, 10))
self.assertEqualArray(d, expected)
# Test num equal to 1
d = mx.linspace(1, 10, 1)
expected = mx.array(np.linspace(1, 10, 1))
self.assertEqualArray(d, expected)
def test_repeat(self):
# Setup data for the tests
data = mx.array([[[13, 3], [16, 6]], [[14, 4], [15, 5]], [[11, 1], [12, 2]]])