#482 need add pyannote-bundle (1h)

482_WhisperX/WhisperX.eb

@@ -12,17 +12,32 @@ toolchain = {'name': 'foss', 'version': '2023a'}
 # DEPS:
     # OK torch>=2
     # OK torchaudio>=2
-    # OK? faster-whisper==1.0.0
+    # OK faster-whisper==1.0.0
     # OK CREATED ctranslate2==4.4.0
     # OK transformers
     # OK pandas
     # OK setuptools>=65
     # OK nltk
+    # ok pyannote.audio==3.1.1
 
 # EXTS:
-    # !!! whisperx 3.2.0 has requirement torchaudio>=2, but you have torchaudio 0.16.2.
-
-
+    # !!! whisperx 3.2.0 has requirement torchaudio>=2, but you have torchaudio 0.16.2. -> E1
+    # pyannote-audio 3.1.1 requires asteroid-filterbanks, which is not installed.
+    # pyannote-audio 3.1.1 requires einops, which is not installed.
+    # pyannote-audio 3.1.1 requires lightning, which is not installed.
+    # pyannote-audio 3.1.1 requires omegaconf, which is not installed.
+    # pyannote-audio 3.1.1 requires pyannote-core, which is not installed.
+    # pyannote-audio 3.1.1 requires pyannote-database, which is not installed.
+    # pyannote-audio 3.1.1 requires pyannote-metrics, which is not installed.
+    # pyannote-audio 3.1.1 requires pyannote-pipeline, which is not installed.
+    # pyannote-audio 3.1.1 requires pytorch-metric-learning, which is not installed.
+    # pyannote-audio 3.1.1 requires semver, which is not installed.
+    # pyannote-audio 3.1.1 requires soundfile, which is not installed.
+    # pyannote-audio 3.1.1 requires speechbrain, which is not installed.
+    # pyannote-audio 3.1.1 requires tensorboardx, which is not installed.
+    # pyannote-audio 3.1.1 requires torch-audiomentations, which is not installed.
+    # pyannote-audio 3.1.1 requires torchmetrics, which is not installed.
+    # pyannote-audio 3.1.1 has requirement torchaudio>=2.0.0, but you have torchaudio 0.16.2.
 
 dependencies = [
     ('CUDA', '12.1.1', '', SYSTEM),
@@ -35,6 +50,7 @@ dependencies = [
     ('FFmpeg', '6.0'),
     ('PyAV', '11.0.0'),
     ('ONNX-Runtime', '1.19.2', versionsuffix),
+    # pyannote-bundle: https://github.com/easybuilders/easybuild-easyconfigs/pull/19992/
 ]
 
 use_pip = True

482_WhisperX/pyannote-bundle.eb

@@ -0,0 +1,61 @@
+easyblock = 'PythonBundle'
+
+name = 'pyannote'
+version = '2025.01'
+local_pytorch_version = '2.1.2'
+local_cuda_version = '-CUDA-%(cudaver)s'
+versionsuffix = '-PyTorch-' + local_pytorch_version + local_cuda_version
+
+
+homepage = 'https://github.com/pyannote'
+description = "pyannote is an open-source toolkit for speaker diarization."
+
+toolchain = {'name': 'foss', 'version': '2023a'}
+
+dependencies = [
+    ('CUDA', '12.1.1', '', SYSTEM),
+    ('Python', '3.11.3'),
+    ('PyTorch-bundle', '2.1.2', local_cuda_version),
+    # ('PyTorch-Lightning', '2.2.0.post0', '-CUDA-%(cudaver)s'),
+    # ('SciPy-bundle', '2023.07'),
+    # ('scikit-learn', '1.3.1'),
+    # ('PyYAML', '6.0'),
+    # ('Optuna', '3.5.0'),
+    # ('matplotlib', '3.7.2'),
+    # ('sympy', '1.12'),
+    # ('typer', '0.9.0'),
+    # ('einops', '0.7.0'),
+    # ('torchaudio', '2.2.0', versionsuffix),
+    # ('torch-audiomentations', '0.11.1', versionsuffix),
+    # ('huggingface-hub', '0.20.3'),
+    # ('asteroid-filterbanks', '0.4.0', versionsuffix),
+    # ('speechbrain', '0.5.16', versionsuffix),
+    # ('pytorch-metric-learning', '2.4.1', versionsuffix),
+    # ('semver', '3.0.2'),
+    # ('omegaconf', '2.3.0', '-Java-11'),
+    # ('lightning', '2.2.0.post0', versionsuffix),
+]
+
+use_pip = True
+
+exts_list = [
+    # ('pyannote.core', '5.0.0', {
+    #     'checksums': ['1a55bcc8bd680ba6be5fa53efa3b6f3d2cdd67144c07b6b4d8d66d5cb0d2096f'],
+    # }),
+    # ('pyannote.database', '5.0.1', {
+    #     'checksums': ['980bdc910615380a17d8bd82805f9770006d050b2c7a946a520ade4627192681'],
+    # }),
+    # ('pyannote.metrics', '3.2.1', {
+    #     'checksums': ['08024255a3550e96a8e9da4f5f4af326886548480de891414567c8900920ee5c'],
+    # }),
+    # ('pyannote.pipeline', '3.0.1', {
+    #     'checksums': ['021794e26a2cf5d8fb5bb1835951e71f5fac33eb14e23dfb7468e16b1b805151'],
+    # }),
+    ('pyannote.audio', '3.1.1', {
+        'checksums': ['b6562d46b5d5d616c1887cafd5aab3f7d19ecf6b3312fc6c4f72f37332fcd408'],
+    }),
+]
+
+sanity_pip_check = True
+
+moduleclass = 'tools'