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

override class function (#1418)

Browse Source
This commit is contained in:
Awni Hannun 2024-09-16 13:21:04 -07:00 committed by GitHub
parent 669c27140d
commit d5ed4d7a71
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
python/mlx/_reprlib_fix.py
View File

@ -3,18 +3,14 @@
import array
import reprlib
_old_repr_array = reprlib.Repr.repr_array
class FixedRepr(reprlib.Repr):
"""Only route python array instances to repr_array."""
def repr_array(self, x, maxlevel):
if isinstance(x, array.array):
return super().repr_array(x, maxlevel)
return _old_repr_array(self, x, maxlevel)
else:
return self.repr_instance(x, maxlevel)
# We need to monkey-patch reprlib so that we can use the debugger without
# renaming the array to something else
fixed_repr = FixedRepr()
reprlib.repr = fixed_repr.repr
reprlib.Repr.repr_array = repr_array