Cleaner import, hold of deprecation for now

Forced-Alignment-and-Vowel-Extraction · Apr 9, 2024 · 711f2a9 · 711f2a9
1 parent f86f1c3
commit 711f2a9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/fave_asr/__init__.py b/src/fave_asr/__init__.py
@@ -1 +1 @@
-from fave_asr.fave_asr import *
+from fave_asr.fave_asr import align, align_segments, assign_speakers, diarize, transcribe, transcribe_and_diarize, to_TextGrid
diff --git a/src/fave_asr/fave_asr.py b/src/fave_asr/fave_asr.py
@@ -101,7 +101,6 @@ def diarize(audio_file: str, hf_token: str) -> Dict[str, Any]:
     diarization_result = diarization_pipeline(audio_file)
     return diarization_result
 
-@warnings.deprecated("Redundant with assign_word_speakers")
 def assign_speakers(
     diarization_result: Dict[str, Any], aligned_segments: Dict[str, Any]
 ) -> List[Dict[str, Any]]:
@@ -116,6 +115,8 @@ def assign_speakers(
         A list of dictionaries representing each segment of the transcript, including 
         the start and end times, the spoken text, and the speaker ID.
     """
+    # Considering deprecation
+    # warnings.warn("Redundant with assign_word_speakers", DeprecationWarning)
     result_segments = assign_word_speakers(
         diarization_result, aligned_segments
     )
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		from fave_asr.fave_asr import *
		from fave_asr.fave_asr import align, align_segments, assign_speakers, diarize, transcribe, transcribe_and_diarize, to_TextGrid