Skip to content

Commit

Permalink
fix syntax error expand_right for python 3.10
Browse files Browse the repository at this point in the history
  • Loading branch information
LarsKue committed Jul 9, 2024
1 parent 6c8d1a1 commit 629712a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion bayesflow/utils/tensor_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ def expand_right(x: Tensor, n: int) -> Tensor:
raise ValueError(f"Cannot expand {n} times.")

idx = [...] + [None] * n
return x[*idx]
return x[tuple(idx)]


def expand_right_to(x: Tensor, dim: int) -> Tensor:
Expand Down

0 comments on commit 629712a

Please sign in to comment.