Skip to content

Commit

Permalink
Apply isort and black reformatting
Browse files Browse the repository at this point in the history
Signed-off-by: tango4j <tango4j@users.noreply.github.com>
  • Loading branch information
tango4j committed Nov 29, 2024
1 parent 4c19278 commit 64eaefd
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions nemo/collections/asr/data/audio_to_diar_label.py
Original file line number Diff line number Diff line change
Expand Up @@ -1235,8 +1235,9 @@ def __getitem__(self, index):

audio_signal_length = torch.tensor(audio_signal.shape[0]).long()
audio_signal, audio_signal_length = audio_signal.to(self.device), audio_signal_length.to(self.device)
target_len = self.get_segment_timestamps(duration=session_len_sec,
sample_rate=self.featurizer.sample_rate).to(self.device)
target_len = self.get_segment_timestamps(duration=session_len_sec, sample_rate=self.featurizer.sample_rate).to(
self.device
)
targets = self.parse_rttm_for_targets_and_lens(
rttm_file=sample.rttm_file, offset=offset, duration=session_len_sec, target_len=target_len
).to(self.device)
Expand Down

0 comments on commit 64eaefd

Please sign in to comment.