zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-18 15:28:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added formatter structure and a boolean value formatter (#354)

Browse Source 
* added formatter structure and a boolean value formatter

---------

Co-authored-by: Awni Hannun <awni@apple.com>
This commit is contained in:
toji
2024-01-18 21:19:41 +05:30
committed by GitHub
parent d1fef34138
commit 49a52610b7
4 changed files with 64 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/src/array.cpp
View File

@@ -520,6 +520,9 @@ class ArrayPythonIterator {
};
void init_array(py::module_& m) {
// Set Python print formatting options
mlx::core::global_formatter.capitalize_bool = true;
// Types
py::class_<Dtype>(
m,

2
python/tests/test_array.py
View File

@@ -304,7 +304,7 @@ class TestArray(mlx_tests.MLXTestCase):
def test_array_repr(self):
x = mx.array(True)
self.assertEqual(str(x), "array(true, dtype=bool)")
self.assertEqual(str(x), "array(True, dtype=bool)")
x = mx.array(1)
self.assertEqual(str(x), "array(1, dtype=int32)")
x = mx.array(1.0)