batchalign 0.7.17.post3__tar.gz → 0.7.17.post5__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (119) hide show
  1. {batchalign-0.7.17.post3/batchalign.egg-info → batchalign-0.7.17.post5}/PKG-INFO +10 -2
  2. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/whisper/infer_asr.py +1 -1
  3. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/asr/utils.py +7 -4
  4. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/asr/whisper.py +4 -1
  5. batchalign-0.7.17.post5/batchalign/version +3 -0
  6. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5/batchalign.egg-info}/PKG-INFO +10 -2
  7. batchalign-0.7.17.post3/batchalign/version +0 -3
  8. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/LICENSE +0 -0
  9. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/MANIFEST.in +0 -0
  10. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/README.md +0 -0
  11. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/__init__.py +0 -0
  12. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/__main__.py +0 -0
  13. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/cli/__init__.py +0 -0
  14. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/cli/cli.py +0 -0
  15. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/cli/dispatch.py +0 -0
  16. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/constants.py +0 -0
  17. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/document.py +0 -0
  18. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/errors.py +0 -0
  19. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/formats/__init__.py +0 -0
  20. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/formats/base.py +0 -0
  21. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/formats/chat/__init__.py +0 -0
  22. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/formats/chat/file.py +0 -0
  23. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/formats/chat/generator.py +0 -0
  24. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/formats/chat/lexer.py +0 -0
  25. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/formats/chat/parser.py +0 -0
  26. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/formats/chat/utils.py +0 -0
  27. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/formats/textgrid/__init__.py +0 -0
  28. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/formats/textgrid/file.py +0 -0
  29. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/formats/textgrid/generator.py +0 -0
  30. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/formats/textgrid/parser.py +0 -0
  31. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/__init__.py +0 -0
  32. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/resolve.py +0 -0
  33. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/speaker/__init__.py +0 -0
  34. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/speaker/config.yaml +0 -0
  35. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/speaker/infer.py +0 -0
  36. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/speaker/utils.py +0 -0
  37. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/training/__init__.py +0 -0
  38. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/training/run.py +0 -0
  39. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/training/utils.py +0 -0
  40. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/utils.py +0 -0
  41. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/utterance/__init__.py +0 -0
  42. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/utterance/cantonese_infer.py +0 -0
  43. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/utterance/dataset.py +0 -0
  44. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/utterance/execute.py +0 -0
  45. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/utterance/infer.py +0 -0
  46. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/utterance/prep.py +0 -0
  47. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/utterance/train.py +0 -0
  48. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/wave2vec/__init__.py +0 -0
  49. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/wave2vec/infer_fa.py +0 -0
  50. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/whisper/__init__.py +0 -0
  51. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/whisper/infer_fa.py +0 -0
  52. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/__init__.py +0 -0
  53. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/analysis/__init__.py +0 -0
  54. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/analysis/eval.py +0 -0
  55. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/asr/__init__.py +0 -0
  56. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/asr/num2chinese.py +0 -0
  57. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/asr/rev.py +0 -0
  58. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/asr/whisperx.py +0 -0
  59. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/base.py +0 -0
  60. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/cleanup/__init__.py +0 -0
  61. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/cleanup/cleanup.py +0 -0
  62. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/cleanup/disfluencies.py +0 -0
  63. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/cleanup/parse_support.py +0 -0
  64. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/cleanup/retrace.py +0 -0
  65. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/cleanup/support/filled_pauses.eng +0 -0
  66. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/cleanup/support/replacements.eng +0 -0
  67. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/cleanup/support/test.test +0 -0
  68. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/dispatch.py +0 -0
  69. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/fa/__init__.py +0 -0
  70. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/fa/wave2vec_fa.py +0 -0
  71. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/fa/whisper_fa.py +0 -0
  72. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/morphosyntax/__init__.py +0 -0
  73. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/morphosyntax/coref.py +0 -0
  74. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/morphosyntax/en/irr.py +0 -0
  75. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/morphosyntax/fr/apm.py +0 -0
  76. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/morphosyntax/fr/apmn.py +0 -0
  77. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/morphosyntax/fr/case.py +0 -0
  78. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/morphosyntax/ja/verbforms.py +0 -0
  79. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/morphosyntax/ud.py +0 -0
  80. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/pipeline.py +0 -0
  81. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/speaker/__init__.py +0 -0
  82. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/speaker/nemo_speaker.py +0 -0
  83. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/translate/__init__.py +0 -0
  84. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/translate/seamless.py +0 -0
  85. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/utr/__init__.py +0 -0
  86. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/utr/rev_utr.py +0 -0
  87. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/utr/utils.py +0 -0
  88. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/utr/whisper_utr.py +0 -0
  89. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/utterance/__init__.py +0 -0
  90. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/utterance/ud_utterance.py +0 -0
  91. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/tests/__init__.py +0 -0
  92. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/tests/conftest.py +0 -0
  93. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/tests/formats/chat/test_chat_file.py +0 -0
  94. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/tests/formats/chat/test_chat_generator.py +0 -0
  95. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/tests/formats/chat/test_chat_lexer.py +0 -0
  96. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/tests/formats/chat/test_chat_parser.py +0 -0
  97. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/tests/formats/chat/test_chat_utils.py +0 -0
  98. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/tests/formats/textgrid/test_textgrid.py +0 -0
  99. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/tests/pipelines/analysis/test_eval.py +0 -0
  100. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/tests/pipelines/asr/test_asr_pipeline.py +0 -0
  101. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/tests/pipelines/asr/test_asr_utils.py +0 -0
  102. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/tests/pipelines/cleanup/test_disfluency.py +0 -0
  103. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/tests/pipelines/cleanup/test_parse_support.py +0 -0
  104. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/tests/pipelines/fa/test_fa_pipeline.py +0 -0
  105. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/tests/pipelines/fixures.py +0 -0
  106. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/tests/pipelines/test_pipeline.py +0 -0
  107. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/tests/pipelines/test_pipeline_models.py +0 -0
  108. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/tests/test_document.py +0 -0
  109. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/utils/__init__.py +0 -0
  110. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/utils/config.py +0 -0
  111. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/utils/dp.py +0 -0
  112. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/utils/utils.py +0 -0
  113. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign.egg-info/SOURCES.txt +0 -0
  114. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign.egg-info/dependency_links.txt +0 -0
  115. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign.egg-info/entry_points.txt +0 -0
  116. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign.egg-info/requires.txt +0 -0
  117. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign.egg-info/top_level.txt +0 -0
  118. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/setup.cfg +0 -0
  119. {batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/setup.py +0 -0
{batchalign-0.7.17.post3/batchalign.egg-info → batchalign-0.7.17.post5}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
- Metadata-Version: 2.1
1
+ Metadata-Version: 2.2
2
2
  Name: batchalign
3
- Version: 0.7.17.post3
3
+ Version: 0.7.17.post5
4
4
  Summary: Python Speech Language Sample Analysis
5
5
  Author: Brian MacWhinney, Houjun Liu
6
6
  Author-email: macw@cmu.edu, houjun@cmu.edu
@@ -40,6 +40,14 @@ Provides-Extra: dev
40
40
  Requires-Dist: pytest; extra == "dev"
41
41
  Provides-Extra: train
42
42
  Requires-Dist: accelerate~=0.27; extra == "train"
43
+ Dynamic: author
44
+ Dynamic: author-email
45
+ Dynamic: classifier
46
+ Dynamic: description
47
+ Dynamic: description-content-type
48
+ Dynamic: provides-extra
49
+ Dynamic: requires-dist
50
+ Dynamic: summary
43
51
 
44
52
  # TalkBank | Batchalign2
45
53
 
{batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/models/whisper/infer_asr.py RENAMED
@@ -67,7 +67,7 @@ class WhisperASRModel(object):
67
67
  L.debug("Initializing whisper model...")
68
68
  self.__config = GenerationConfig.from_pretrained(base)
69
69
  self.__config.no_repeat_ngram_size = 4
70
-
70
+ self.__config.use_cache = True
71
71
 
72
72
  if language == "Cantonese":
73
73
  self.__config.no_repeat_ngram_size = 4
{batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/asr/utils.py RENAMED
@@ -98,10 +98,13 @@ def retokenize_with_engine(intermediate_output, engine):
98
98
  tmp = []
99
99
 
100
100
  for s in new_ut:
101
- try:
102
- tmp.append((s, utterance.pop(0)[1]))
103
- except IndexError:
104
- continue
101
+ if current_index < len(utterance):
102
+ # Use current element and move index forward
103
+ tmp.append((s, utterance[current_index][1]))
104
+ current_index += 1
105
+ else:
106
+ # Append with default timestamp if utterance is exhausted
107
+ tmp.append((s, [None, None]))
105
108
 
106
109
  final_outputs.append((speaker, tmp+[[delim, [None, None]]]))
107
110
 
{batchalign-0.7.17.post3 → batchalign-0.7.17.post5}/batchalign/pipelines/asr/whisper.py RENAMED
@@ -33,7 +33,10 @@ class WhisperEngine(BatchalignEngine):
33
33
  model = "openai/whisper-large-v3"
34
34
  base = "openai/whisper-large-v3"
35
35
 
36
- language = pycountry.languages.get(alpha_3=lang).name
36
+ if lang == "mys":
37
+ language = "Malay"
38
+ else:
39
+ language = pycountry.languages.get(alpha_3=lang).name
37
40
  if language == "Yue Chinese":
38
41
  language = "Cantonese"
39
42
  if "greek" in language.lower():
batchalign-0.7.17.post5/batchalign/version ADDED
@@ -0,0 +1,3 @@
1
+ 0.7.17-post.5
2
+ March 10th, 2025
3
+ more whisper changes
{batchalign-0.7.17.post3 → batchalign-0.7.17.post5/batchalign.egg-info}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
- Metadata-Version: 2.1
1
+ Metadata-Version: 2.2
2
2
  Name: batchalign
3
- Version: 0.7.17.post3
3
+ Version: 0.7.17.post5
4
4
  Summary: Python Speech Language Sample Analysis
5
5
  Author: Brian MacWhinney, Houjun Liu
6
6
  Author-email: macw@cmu.edu, houjun@cmu.edu
@@ -40,6 +40,14 @@ Provides-Extra: dev
40
40
  Requires-Dist: pytest; extra == "dev"
41
41
  Provides-Extra: train
42
42
  Requires-Dist: accelerate~=0.27; extra == "train"
43
+ Dynamic: author
44
+ Dynamic: author-email
45
+ Dynamic: classifier
46
+ Dynamic: description
47
+ Dynamic: description-content-type
48
+ Dynamic: provides-extra
49
+ Dynamic: requires-dist
50
+ Dynamic: summary
43
51
 
44
52
  # TalkBank | Batchalign2
45
53
 
batchalign-0.7.17.post3/batchalign/version DELETED
@@ -1,3 +0,0 @@
1
- 0.7.17-post.3
2
- March 7th, 2025
3
- bfloat16 inference for whisper