TBNilles/LEANN
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

style: Apply ruff formatting to embedding_compute.py

Browse Source 
- Break long logger.warning lines for better readability
- Fixes pre-commit hook formatting requirements
This commit is contained in:
aakash
2025-10-06 14:54:17 -07:00
parent 4ccbbf3e6b
commit c24e62a3d9
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
packages/leann-core/src/leann/embedding_compute.py
View File

@@ -199,7 +199,9 @@ def compute_embeddings_sentence_transformers(
logger.info("Model loaded successfully! (local + optimized)") logger.info("Model loaded successfully! (local + optimized)")
except TypeError as e: except TypeError as e:
if "model_kwargs" in str(e) or "tokenizer_kwargs" in str(e): if "model_kwargs" in str(e) or "tokenizer_kwargs" in str(e):
logger.warning(f"Advanced parameters not supported ({e}), using basic initialization...") logger.warning(
f"Advanced parameters not supported ({e}), using basic initialization..."
)
# Fallback to basic initialization for older versions # Fallback to basic initialization for older versions
try: try:
model = SentenceTransformer( model = SentenceTransformer(
@@ -237,7 +239,9 @@ def compute_embeddings_sentence_transformers(
logger.info("Model loaded successfully! (network + optimized)") logger.info("Model loaded successfully! (network + optimized)")
except TypeError as e2: except TypeError as e2:
if "model_kwargs" in str(e2) or "tokenizer_kwargs" in str(e2): if "model_kwargs" in str(e2) or "tokenizer_kwargs" in str(e2):
logger.warning(f"Advanced parameters not supported ({e2}), using basic network loading...") logger.warning(
f"Advanced parameters not supported ({e2}), using basic network loading..."
)
model = SentenceTransformer( model = SentenceTransformer(
model_name, model_name,
device=device, device=device,