Skip to content

Commit

Permalink
removing transcribe_timestamped wf #3
Browse files Browse the repository at this point in the history
  • Loading branch information
fabiocat93 committed Sep 19, 2024
1 parent 384ce07 commit ba1ffac
Show file tree
Hide file tree
Showing 3 changed files with 10 additions and 1 deletion.
Original file line number Diff line number Diff line change
@@ -1,5 +1,8 @@
"""Workflow for timestamped transcription."""

"""
# TODO: Please double-check this because tests are failing
from senselab.audio.workflows.transcribe_timestamped.transcribe_timestamped import transcribe_timestamped
__all__ = ["transcribe_timestamped"]
"""
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
"""Transcribes audio files with timestamps."""

'''
# TODO: Please double-check this because tests are failing
from typing import List
import pydra
Expand Down Expand Up @@ -77,7 +79,7 @@ def transcribe_task(audios: List[Audio], model: HFModel, language: Language) ->
model=wf.lzin.model,
language=wf.lzin.language,
)
).split("batched_audios", batched_audios=batched_audios)
).split("batched_audios", batched_audios=wf.transcribe.lzin.batched_audios)
align_transcriptions_task = pydra.mark.task(align_transcriptions)
wf.add(
Expand All @@ -99,3 +101,4 @@ def transcribe_task(audios: List[Audio], model: HFModel, language: Language) ->
sub(wf)
return wf.result()[0].output.aligned_transcriptions
'''
3 changes: 3 additions & 0 deletions src/tests/audio/workflows/transcribe_timestamped_test.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
"""Tests the transcribe_timestamped module."""

'''
# TODO: Please double-check this because tests are failing
from senselab.audio.data_structures.audio import Audio
from senselab.audio.workflows.transcribe_timestamped import transcribe_timestamped
Expand Down Expand Up @@ -70,3 +72,4 @@ def test_transcribe_timestamped_different_bit_depths(
assert all(
len(script_lines) > 0 for script_lines in result
), "Each list in the result should contain ScriptLine objects."
'''

0 comments on commit ba1ffac

Please sign in to comment.