Skip to content

Commit 61a46fe

Browse files
authored
Add explicit dtypes for cumsum (#94)
1 parent 46fe27b commit 61a46fe

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

i6_models/parts/rasr_fsa.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -322,8 +322,8 @@ def join_fsas_fbw_v2(fsas: Iterable[FsaTuple]) -> WeightedFsaV2:
322322
fsas = list(fsas) # ensure we can iterate multiple times over this iterable
323323
num_states = [f[0] for f in fsas]
324324
num_edges = [f[1] for f in fsas]
325-
start_states = np.cumsum(np.array([0] + num_states, dtype=np.uint32))[:-1]
326-
end_states = np.cumsum(num_states) - 1
325+
start_states = np.cumsum(np.array([0] + num_states), dtype=np.uint32)[:-1]
326+
end_states = np.cumsum(num_states, dtype=np.uint32) - 1
327327
weights = np.concatenate(tuple(f[3] for f in fsas))
328328

329329
edges = []

0 commit comments

Comments
 (0)