






'''simple docstring'''







import copy
from dataclasses import dataclass, field
from typing import ClassVar, Dict

from ..features import Audio, Features, Value
from .base import TaskTemplate




@dataclass(frozen=a			)
class 		UpperCamelCase__    (a			):




		'''simple docstring'''





		_UpperCamelCase				  =					field(default='automatic-speech-recognition'   ,metadata={'include_in_asdict_even_if_is_default': True}			)
		_UpperCamelCase				  =					Features({'audio': Audio()}			)
		_UpperCamelCase				  =					Features({'transcription': Value('string'			)}			)
		_UpperCamelCase				  =					'audio'
		_UpperCamelCase				  =					'transcription'






		def 			UpperCamelCase_					(      self   ,_lowerCAmelCase       ):
								if self.audio_column not in features:
														raise ValueError(F'''Column {self.audio_column} is not present in features.'''       )
								if not isinstance(features[self.audio_column]   ,_lowerCAmelCase       ):
														raise ValueError(F'''Column {self.audio_column} is not an Audio type.'''       )
								lowerCamelCase__         =  copy.deepcopy(self       )
								lowerCamelCase__         =  self.input_schema.copy()
								lowerCamelCase__         =  features[self.audio_column]
								lowerCamelCase__         =  input_schema
								return task_template






		@property
		def 			UpperCamelCase_					(      self       ):
								return {self.audio_column: "audio", self.transcription_column: "transcription"}









'''simple docstring'''







from manim import *




class 		UpperCamelCase__    (a			):




		'''simple docstring'''





		def 			UpperCamelCase_					(      self       ):
								lowerCamelCase__         =  Rectangle(height=0.5   ,width=0.5       )
								lowerCamelCase__         =  Rectangle(height=0.46   ,width=0.46       ).set_stroke(width=0       )

								lowerCamelCase__         =  [mem.copy() for i in range(6       )]
								lowerCamelCase__         =  [mem.copy() for i in range(6       )]
								lowerCamelCase__         =  VGroup(*_lowerCAmelCase       ).arrange(_lowerCAmelCase   ,buff=0       )
								lowerCamelCase__         =  VGroup(*_lowerCAmelCase       ).arrange(_lowerCAmelCase   ,buff=0       )
								lowerCamelCase__         =  VGroup(_lowerCAmelCase   ,_lowerCAmelCase       ).arrange(_lowerCAmelCase   ,buff=0       )
								lowerCamelCase__         =  Text("""CPU"""   ,font_size=24       )
								lowerCamelCase__         =  Group(_lowerCAmelCase   ,_lowerCAmelCase       ).arrange(_lowerCAmelCase   ,buff=0.5   ,aligned_edge=_lowerCAmelCase       )
								cpu.move_to([-2.5, -0.5, 0]       )
								self.add(_lowerCAmelCase       )

								lowerCamelCase__         =  [mem.copy() for i in range(1       )]
								lowerCamelCase__         =  VGroup(*_lowerCAmelCase       ).arrange(_lowerCAmelCase   ,buff=0       )
								lowerCamelCase__         =  Text("""GPU"""   ,font_size=24       )
								lowerCamelCase__         =  Group(_lowerCAmelCase   ,_lowerCAmelCase       ).arrange(_lowerCAmelCase   ,buff=0.5   ,aligned_edge=_lowerCAmelCase       )
								gpu.align_to(_lowerCAmelCase   ,_lowerCAmelCase       )
								gpu.set_x(gpu.get_x() - 1       )

								self.add(_lowerCAmelCase       )

								lowerCamelCase__         =  [mem.copy() for i in range(6       )]
								lowerCamelCase__         =  VGroup(*_lowerCAmelCase       ).arrange(_lowerCAmelCase   ,buff=0       )

								lowerCamelCase__         =  Text("""Model"""   ,font_size=24       )
								lowerCamelCase__         =  Group(_lowerCAmelCase   ,_lowerCAmelCase       ).arrange(_lowerCAmelCase   ,buff=0.5   ,aligned_edge=_lowerCAmelCase       )
								model.move_to([3, -1.0, 0]       )

								self.play(
								    Create(_lowerCAmelCase   ,run_time=1       )   ,Create(_lowerCAmelCase   ,run_time=1       )   ,Create(_lowerCAmelCase   ,run_time=1       )   ,)

								lowerCamelCase__         =  MarkupText(
								    F'''First, an empty model skeleton is loaded\ninto <span fgcolor=\'{YELLOW}\'>memory</span> without using much RAM.'''   ,font_size=24   ,)

								lowerCamelCase__         =  Square(side_length=2.2       )
								key.move_to([-5, 2, 0]       )

								lowerCamelCase__         =  MarkupText(
								    F'''<b>Key:</b>\n\n<span fgcolor=\'{YELLOW}\'>●</span> Empty Model'''   ,font_size=18   ,)

								key_text.move_to([-5, 2.4, 0]       )

								step_a.move_to([2, 2, 0]       )
								self.play(Write(_lowerCAmelCase   ,run_time=2.5       )   ,Write(_lowerCAmelCase       )   ,Write(_lowerCAmelCase       )       )

								self.add(_lowerCAmelCase       )

								lowerCamelCase__         =  []
								lowerCamelCase__         =  []
								lowerCamelCase__         =  []
								for i, rect in enumerate(_lowerCAmelCase       ):
														lowerCamelCase__         =  Rectangle(height=0.46   ,width=0.46       ).set_stroke(width=0.0       ).set_fill(_lowerCAmelCase   ,opacity=0.7       )
														cpu_target.move_to(_lowerCAmelCase       )
														cpu_target.generate_target()
														lowerCamelCase__         =  0.46 / 4
														lowerCamelCase__         =  0.46 / 3

														if i == 0:
																				cpu_target.target.next_to(cpu_left_col_base[0].get_corner(DOWN + LEFT       )   ,buff=0.02   ,direction=_lowerCAmelCase       )
																				cpu_target.target.set_x(cpu_target.target.get_x() + 0.1       )
														elif i == 3:
																				cpu_target.target.next_to(cpu_targs[0].target   ,direction=_lowerCAmelCase   ,buff=0.0       )
														else:
																				cpu_target.target.next_to(cpu_targs[i - 1].target   ,direction=_lowerCAmelCase   ,buff=0.0       )
														cpu_targs.append(_lowerCAmelCase       )

														first_animations.append(rect.animate(run_time=0.5       ).set_stroke(_lowerCAmelCase       )       )
														second_animations.append(MoveToTarget(_lowerCAmelCase   ,run_time=1.5       )       )

								self.play(*_lowerCAmelCase       )
								self.play(*_lowerCAmelCase       )

								self.wait()

