mirror of
https://github.com/Monadical-SAS/reflector.git
synced 2026-02-04 09:56:47 +00:00
Fix: Preserve track_index explicitly in PaddedTrackInfo
- Add track_index to PaddedTrackInfo model - Preserve track_index from PadTrackResult when building padded_tracks list - Use explicit track_index instead of enumerate in process_transcriptions - Removes fragile ordering assumption
This commit is contained in:
@@ -434,11 +434,12 @@ async def process_paddings(input: PipelineInput, ctx: Context) -> ProcessPadding
|
|||||||
for result in results:
|
for result in results:
|
||||||
pad_result = PadTrackResult(**result[TaskName.PAD_TRACK])
|
pad_result = PadTrackResult(**result[TaskName.PAD_TRACK])
|
||||||
|
|
||||||
# Store S3 key info (not presigned URL) - consumer tasks presign on demand
|
|
||||||
if pad_result.padded_key:
|
if pad_result.padded_key:
|
||||||
padded_tracks.append(
|
padded_tracks.append(
|
||||||
PaddedTrackInfo(
|
PaddedTrackInfo(
|
||||||
key=pad_result.padded_key, bucket_name=pad_result.bucket_name
|
key=pad_result.padded_key,
|
||||||
|
bucket_name=pad_result.bucket_name,
|
||||||
|
track_index=pad_result.track_index,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -482,13 +483,13 @@ async def process_transcriptions(
|
|||||||
bulk_runs = [
|
bulk_runs = [
|
||||||
transcription_workflow.create_bulk_run_item(
|
transcription_workflow.create_bulk_run_item(
|
||||||
input=TranscriptionInput(
|
input=TranscriptionInput(
|
||||||
track_index=i,
|
track_index=padded_track.track_index,
|
||||||
padded_key=padded_track.key,
|
padded_key=padded_track.key,
|
||||||
bucket_name=padded_track.bucket_name,
|
bucket_name=padded_track.bucket_name,
|
||||||
language=source_language,
|
language=source_language,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
for i, padded_track in enumerate(padded_tracks)
|
for padded_track in padded_tracks
|
||||||
]
|
]
|
||||||
|
|
||||||
results = await transcription_workflow.aio_run_many(bulk_runs)
|
results = await transcription_workflow.aio_run_many(bulk_runs)
|
||||||
|
|||||||
@@ -58,6 +58,7 @@ class PaddedTrackInfo(BaseModel):
|
|||||||
|
|
||||||
key: NonEmptyString
|
key: NonEmptyString
|
||||||
bucket_name: NonEmptyString | None
|
bucket_name: NonEmptyString | None
|
||||||
|
track_index: int
|
||||||
|
|
||||||
|
|
||||||
class ProcessPaddingsResult(BaseModel):
|
class ProcessPaddingsResult(BaseModel):
|
||||||
|
|||||||
Reference in New Issue
Block a user