Skip to content

Commit 03d236c

Browse files
committed
Added check for total words
1 parent 44f1b39 commit 03d236c

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

app/training_processor.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
from app.training import Training
1313

1414
logger = get_root_logger(service_name='training_processor')
15-
15+
TOTAL_WORDS_CRITERIA = 1e10
1616

1717
class TrainingProcessor:
1818
def run(self):
@@ -60,6 +60,16 @@ def run(self):
6060
continue
6161
presentation = Presentation.from_json_file(presentation_file)
6262
presentation_file.close()
63+
if audio.audio_stats['total_words'] < TOTAL_WORDS_CRITERIA * audio.audio_stats['duration']:
64+
TrainingsDBManager().change_training_status_by_training_id(
65+
training_id, TrainingStatus.PROCESSING_FAILED
66+
)
67+
verdict = 'Not enough words, total words = {}, duration = {}, training_id = {}.'\
68+
.format(audio.audio_stats['total_words'], audio.audio_stats['duration'], training_id)
69+
TrainingsDBManager().append_verdict(training_id, verdict)
70+
TrainingsDBManager().set_score(training_id, 0)
71+
logger.warning(verdict)
72+
continue
6373
criteria_pack_id = training_db.criteria_pack_id
6474
criteria_pack = CriteriaPackFactory().get_criteria_pack(criteria_pack_id)
6575
criteria_pack_db = CriterionPackDBManager().get_criterion_pack_by_name(criteria_pack.name)

0 commit comments

Comments
 (0)