mirror of
https://github.com/ml-explore/mlx.git
synced 2025-10-18 15:28:16 +08:00
Added formatter structure and a boolean value formatter (#354)
* added formatter structure and a boolean value formatter --------- Co-authored-by: Awni Hannun <awni@apple.com>
This commit is contained in:
@@ -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,
|
||||
|
@@ -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)
|
||||
|
Reference in New Issue
Block a user