From 6cdcd2759c727cd7f977487f9456c96b1952f543 Mon Sep 17 00:00:00 2001 From: Pablosinyores Date: Fri, 3 Jul 2026 13:28:25 +0530 Subject: [PATCH] Fix Conv3d docstring argument order (padding before dilation) The Conv3d docstring listed the constructor arguments as stride, dilation, padding, but the signature order is stride, padding, dilation (matching Conv1d and Conv2d). A caller following the docstring with positional arguments would swap padding and dilation. Reorder the docstring to match the signature and the Conv1d/Conv2d docstrings. --- python/mlx/nn/layers/convolution.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/mlx/nn/layers/convolution.py b/python/mlx/nn/layers/convolution.py index 82763868d5..e7396008e7 100644 --- a/python/mlx/nn/layers/convolution.py +++ b/python/mlx/nn/layers/convolution.py @@ -179,9 +179,9 @@ class Conv3d(Module): kernel_size (int or tuple): The size of the convolution filters. stride (int or tuple, optional): The size of the stride when applying the filter. Default: ``1``. - dilation (int or tuple, optional): The dilation of the convolution. padding (int or tuple, optional): How many positions to 0-pad the input with. Default: ``0``. + dilation (int or tuple, optional): The dilation of the convolution. bias (bool, optional): If ``True`` add a learnable bias to the output. Default: ``True`` """