switch to non-interactive outputs

Author: satra

.github/workflows/ci.yml

@@ -36,4 +36,4 @@ jobs:
           pip install --upgrade --force-reinstall --no-cache-dir --editable=".[test]"
       - name: Test with pytest
         run: |
-          pytest -sv pydra_ml/tests/test_classifier.py
+          pytest pydra_ml/tests/test_classifier.py

pydra_ml/report.py

@@ -3,12 +3,15 @@
 import pickle
 import warnings
 
+import matplotlib
 import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
 import seaborn as sns
 from sklearn.metrics import accuracy_score, explained_variance_score
 
+matplotlib.use("Agg")
+
 
 def save_obj(obj, path):
     with open(path, "wb") as f:

pydra_ml/tasks.py

@@ -254,7 +254,9 @@ def get_shap(X, permute, model, gen_shap=False, nsamples="auto", l1_reg="aic"):
     import shap
 
     explainer = shap.KernelExplainer(pipe.predict, shap.kmeans(X[train_index], 5))
-    shaps = explainer.shap_values(X[test_index], nsamples=nsamples, l1_reg=l1_reg)
+    shaps = explainer.shap_values(
+        X[test_index], nsamples=nsamples, l1_reg=l1_reg, silent=True
+    )
     return shaps