Skip to content

Commit

Permalink
update typing
Browse files Browse the repository at this point in the history
Co-authored-by: michelwi <[email protected]>
  • Loading branch information
DanEnergetics and michelwi authored Aug 26, 2024
1 parent 43b19b2 commit af9b317
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion i6_models/parts/fsa.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ def build_single(self, seq_tag: str) -> Tuple[int, int, np.ndarray, np.ndarray]:
raw_fsa = self.builder.build_by_segment_name(seq_tag)
return raw_fsa

def build_batch(self, seq_tags: Iterable[str]) -> TWeightedFsa:
def build_batch(self, seq_tags: Iterable[str]) -> WeightedFsa:
"""
Build and concatenate the FSAs for a batch of sequence tags
and reformat as an input to `i6_native_ops.fbw.fbw_loss`.
Expand Down

0 comments on commit af9b317

Please sign in to comment.