mlx/python/mlx/_reprlib_fix.py

# Copyright © 2023 Apple Inc.

import array
import reprlib

_old_repr_array = reprlib.Repr.repr_array


def repr_array(self, x, maxlevel):
    if isinstance(x, array.array):
        return _old_repr_array(self, x, maxlevel)
    else:
        return self.repr_instance(x, maxlevel)


reprlib.Repr.repr_array = repr_array
copyright + ack 2023-12-01 03:12:53 +08:00			`# Copyright © 2023 Apple Inc.`

awni's commit files 2023-11-30 02:30:41 +08:00			`import array`
			`import reprlib`

override class function (#1418) 2024-09-17 04:21:04 +08:00			`_old_repr_array = reprlib.Repr.repr_array`
awni's commit files 2023-11-30 02:30:41 +08:00

override class function (#1418) 2024-09-17 04:21:04 +08:00			`def repr_array(self, x, maxlevel):`
			`if isinstance(x, array.array):`
			`return _old_repr_array(self, x, maxlevel)`
			`else:`
			`return self.repr_instance(x, maxlevel)`
awni's commit files 2023-11-30 02:30:41 +08:00

override class function (#1418) 2024-09-17 04:21:04 +08:00			`reprlib.Repr.repr_array = repr_array`