def shap_predict_wrapper()

in src/infer_location.py [0:0]

• 5 lines of code
• 3 McCabe index (conditional complexity)

    def shap_predict_wrapper(self, texts):
        if isinstance(texts, np.ndarray):
            texts = texts.flatten().tolist()  # Ensure it’s a list of strings
        probs = self.predict_proba(texts)
        # Flatten output to ensure it's a 2D array
        return np.array([p.flatten() for p in probs])