Skip to content

Commit

Permalink
CLN: Type the return of determine_adv_broadcasting
Browse files Browse the repository at this point in the history
  • Loading branch information
Jacob-Stevens-Haas committed Jan 4, 2024
1 parent 91064dd commit 700521a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pysindy/utils/axes.py
Original file line number Diff line number Diff line change
Expand Up @@ -481,7 +481,7 @@ def _move_idxs_to_front(li: list, idxs: Sequence) -> None:

def _determine_adv_broadcasting(
key: StandardIndexer | Sequence[StandardIndexer], adv_inds: Sequence[OldIndex]
) -> tuple:
) -> tuple[bool, int, Optional[int]]:
"""Calculate the shape and location for the result of advanced indexing."""
adjacent = all(i + 1 == j for i, j in zip(adv_inds[:-1], adv_inds[1:]))
adv_indexers = [np.array(key[i]) for i in adv_inds]
Expand Down

0 comments on commit 700521a

Please sign in to comment.