From 5e691ef779e84b34199e04af1f2ea5f16405366f Mon Sep 17 00:00:00 2001 From: Jackson McClintock Date: Wed, 10 Apr 2024 16:52:53 -0400 Subject: [PATCH] fix: correct type in ivy.experimental.manipulation `fill_diagonal` --- ivy/functional/ivy/experimental/manipulation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ivy/functional/ivy/experimental/manipulation.py b/ivy/functional/ivy/experimental/manipulation.py index a1467eddf7271..0aa7400b30856 100644 --- a/ivy/functional/ivy/experimental/manipulation.py +++ b/ivy/functional/ivy/experimental/manipulation.py @@ -2203,7 +2203,7 @@ def fill_diagonal( steps = ivy.arange(0, end, step) if isinstance(v, (ivy.Array, ivy.NativeArray)): v = ivy.reshape(v, (-1,)).astype(a.dtype) - v = ivy.tile(v, int(ivy.ceil(len(steps) / v.shape[0])))[: len(steps)] + v = ivy.tile(v, (int(ivy.ceil(len(steps) / v.shape[0])),))[: len(steps)] else: v = ivy.repeat(v, len(steps)) ivy.scatter_flat(steps, v, size=a.shape[0], reduction="replace", out=a)