Fix linting errors in colqwen_rag.py and test_colqwen_reproduction.py

- Add noqa comments for E402 errors (imports after sys.path modifications)
- Remove unused variable assignment in colqwen_rag.py
- Use importlib.util.find_spec for dependency checks instead of unused imports
- Fix import ordering in test_colqwen_reproduction.py
This commit is contained in:
aakash
2025-11-11 05:12:49 -08:00
parent 9dd0e0b26f
commit 9b7353f336
4 changed files with 19 additions and 13 deletions

View File

@@ -11,6 +11,7 @@ This script demonstrates the ColQwen workflow:
5. Generate similarity maps
"""
import importlib.util
import os
from pathlib import Path
@@ -31,9 +32,14 @@ def main():
# Step 1: Check dependencies
print("\n📦 Checking dependencies...")
try:
import pdf2image
import torch
from colpali_engine.models import ColQwen2
# Check if pdf2image is available
if importlib.util.find_spec("pdf2image") is None:
raise ImportError("pdf2image not found")
# Check if colpali_engine is available
if importlib.util.find_spec("colpali_engine") is None:
raise ImportError("colpali_engine not found")
print("✅ Core dependencies available")
print(f" - PyTorch: {torch.__version__}")