Skip to content
Snippets Groups Projects
Select Git revision
  • 0a9519f11de73bce2e1ba11b7d0b79bf1eacfc4f
  • main default protected
  • change_data_model
  • feature/add_auth_asr_service
  • fix/incorrect_import
  • feature/change_registry_clarin
  • feature/add_base_asr_service
  • feature/add_poetry
  • feature/add_word_ids
  • feature/add_sziszapangma
10 results

gold_transcript_task.py

Blame
  • user avatar
    Marcin Wątroba authored
    0a9519f1
    History
    gold_transcript_task.py 1.45 KiB
    from sziszapangma.integration.gold_transcript_processor import \
        GoldTranscriptProcessor
    from sziszapangma.integration.record_path_provider import RecordPathProvider
    from sziszapangma.integration.repository.experiment_repository \
        import ExperimentRepository
    from sziszapangma.integration.task.processing_task import ProcessingTask
    
    
    class GoldTranscriptTask(ProcessingTask):
        _gold_transcript_processor: GoldTranscriptProcessor
        _gold_transcript_property_name: str
    
        def __init__(
            self,
            task_name: str,
            gold_transcript_processor: GoldTranscriptProcessor,
            gold_transcript_property_name: str,
            require_update: bool
        ):
            super().__init__(task_name, require_update)
            self._gold_transcript_processor = gold_transcript_processor
            self._gold_transcript_property_name = gold_transcript_property_name
    
        def skip_for_record(self, record_id: str,
                            experiment_repository: ExperimentRepository) -> bool:
            return experiment_repository \
                .get_property_for_key(record_id,
                                      self._gold_transcript_property_name)
    
        def run_single_process(self, record_id: str,
                               experiment_repository: ExperimentRepository):
            experiment_repository.update_property_for_key(
                record_id,
                self._gold_transcript_property_name,
                self._gold_transcript_processor.get_gold_transcript(record_id)
            )