Skip to content

Commit

Permalink
fix: correct type in ivy.experimental.manipulation fill_diagonal
Browse files Browse the repository at this point in the history
  • Loading branch information
jacksondm33 committed Apr 12, 2024
1 parent c91b9d7 commit 41228de
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ivy/functional/ivy/experimental/manipulation.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 41228de

Please sign in to comment.