From cffceda6ee6d5dcd91bae28741fd858699dbe67d Mon Sep 17 00:00:00 2001 From: Chunyang Wen Date: Mon, 10 Mar 2025 21:05:36 +0800 Subject: [PATCH] Add type hint for _extra_repr (#1948) --- python/mlx/nn/layers/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/mlx/nn/layers/base.py b/python/mlx/nn/layers/base.py index 3a696df71..b35c58478 100644 --- a/python/mlx/nn/layers/base.py +++ b/python/mlx/nn/layers/base.py @@ -81,7 +81,7 @@ class Module(dict): """ return self - def _extra_repr(self): + def _extra_repr(self) -> str: return "" def __repr__(self):