mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-12-08 20:58:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix unit tests and code coverage for CI

Browse Source
This commit is contained in:
Ken Steele
2025-10-04 03:32:21 -07:00
parent 93f059990b
commit 8371c060ea
1 changed files with 26 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
docs/examples/asr_pipeline_performance_comparison.py vendored
View File

@@ -174,15 +174,15 @@ def main():
print(f"MLX Whisper available: {has_mlx_whisper}") print(f"MLX Whisper available: {has_mlx_whisper}")
if not has_mps: if not has_mps:
print("❌ This test requires Apple Silicon (MPS) to be meaningful.") print("⚠️ Apple Silicon (MPS) not available - running CPU-only comparison")
print(" For MLX Whisper performance benefits, run on Apple Silicon devices")
print(" MLX Whisper is optimized for Apple Silicon devices.") print(" MLX Whisper is optimized for Apple Silicon devices.")
sys.exit(1)
if not has_mlx_whisper: if not has_mlx_whisper:
print("❌ MLX Whisper is not installed.") print("⚠️ MLX Whisper not installed - running CPU-only comparison")
print(" Install with: pip install mlx-whisper") print(" Install with: pip install mlx-whisper")
print(" Or: uv sync --extra asr") print(" Or: uv sync --extra asr")
sys.exit(1) print(" For MLX Whisper performance benefits, install the dependency")
# Determine audio file path # Determine audio file path
if args.audio: if args.audio:
@@ -229,15 +229,21 @@ def main():
) )
model_results["cpu"] = {"duration": cpu_duration, "success": cpu_success} model_results["cpu"] = {"duration": cpu_duration, "success": cpu_success}
# Test 2: MLX Whisper (Apple Silicon optimized) # Test 2: MLX Whisper (Apple Silicon optimized) - only if available
mlx_options = create_mlx_whisper_options(model_size) if has_mps and has_mlx_whisper:
mlx_duration, mlx_success = run_transcription_test( mlx_options = create_mlx_whisper_options(model_size)
audio_file, mlx_duration, mlx_success = run_transcription_test(
mlx_options, audio_file,
AcceleratorDevice.MPS, mlx_options,
f"MLX Whisper {model_size} (MPS)", AcceleratorDevice.MPS,
) f"MLX Whisper {model_size} (MPS)",
model_results["mlx"] = {"duration": mlx_duration, "success": mlx_success} )
model_results["mlx"] = {"duration": mlx_duration, "success": mlx_success}
else:
print(f"\n{'=' * 60}")
print(f"Skipping MLX Whisper {model_size} (MPS) - not available")
print(f"{'=' * 60}")
model_results["mlx"] = {"duration": 0.0, "success": False}
results[model_size] = model_results results[model_size] = model_results
@@ -292,7 +298,13 @@ def main():
print(f"\n🎯 MLX Whisper provides {avg_speedup:.1f}x average speedup over CPU!") print(f"\n🎯 MLX Whisper provides {avg_speedup:.1f}x average speedup over CPU!")
else: else:
print("\n❌ No successful comparisons available.") if has_mps and has_mlx_whisper:
print("\n❌ No successful comparisons available.")
else:
print("\n⚠️ MLX Whisper not available - only CPU results shown.")
print(
" Install MLX Whisper and run on Apple Silicon for performance comparison."
)
if __name__ == "__main__": if __name__ == "__main__":