Skip to content

Commit f1a7280

Browse files
committed
fix: make stylo_metrix import conditional to fix CI
1 parent def08d6 commit f1a7280

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

stylometry/metrics.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,10 @@
44
import numpy as np
55
import pandas as pd
66
from typing import Dict, List, Sequence, Tuple, Optional
7-
from stylo_metrix import StyloMetrix
7+
try:
8+
from stylo_metrix import StyloMetrix
9+
except ImportError:
10+
StyloMetrix = None
811
from sklearn.feature_extraction.text import TfidfVectorizer
912
from sklearn.metrics.pairwise import cosine_similarity
1013
from sklearn.decomposition import PCA
@@ -137,8 +140,12 @@ def compute_metrics_for_text(word_tokens: List[str], sentences: List[str], raw_t
137140

138141
_SM_INSTANCE: Optional[StyloMetrix] = None
139142

140-
def get_sm_instance(lang: str = "en") -> StyloMetrix:
143+
def get_sm_instance(lang: str = "en"):
141144
global _SM_INSTANCE
145+
146+
if StyloMetrix is None:
147+
raise ImportError("StyloMetrix is not installed. Install with 'pip install stylometry-cli[stylometrix]' or 'pip install stylo_metrix' to use this feature.")
148+
142149
if _SM_INSTANCE is None:
143150
try:
144151
_SM_INSTANCE = StyloMetrix(lang)

0 commit comments

Comments
 (0)