Merge pull request #263 from py-why/fix/local-score-numpy-scalar

Fix NumPy scalar extraction in BIC scores

Author: jdramsey

causallearn/score/LocalScoreFunction.py

@@ -84,7 +84,7 @@ def local_score_BIC(Data: ndarray, i: int, PAi: List[int], parameters=None) -> f
             XX_inv = np.linalg.inv(XX)
         except np.linalg.LinAlgError:
             XX_inv = np.linalg.pinv(XX)
-        sigma = float(cov[i, i] - yX @ XX_inv @ yX.T)
+        sigma = float((cov[i, i] - yX @ XX_inv @ yX.T).item())
 
     if sigma <= 0:
         sigma = np.finfo(float).eps
@@ -137,7 +137,7 @@ def local_score_BIC_from_cov(
             XX_inv = np.linalg.inv(XX)
         except np.linalg.LinAlgError:
             XX_inv = np.linalg.pinv(XX)
-        sigma = float(cov[i, i] - yX @ XX_inv @ yX.T)
+        sigma = float((cov[i, i] - yX @ XX_inv @ yX.T).item())
 
     if sigma <= 0:
         sigma = np.finfo(float).eps

causallearn/search/ScoreBased/DGES.py

@@ -110,7 +110,7 @@ def local_score_BIC_from_cov_deterministic(Data, i, PAi, parameters=None):
             XX_inv = np.linalg.inv(XX)
         except np.linalg.LinAlgError:
             XX_inv = np.linalg.pinv(XX)
-        sigma = float(cov[i, i] - yX @ XX_inv @ yX.T)
+        sigma = float((cov[i, i] - yX @ XX_inv @ yX.T).item())
     if sigma <= 0:
         sigma = 0.0