Removed graphviz installation; Used "agg" as mpl backend

neptune-ai · Mar 26, 2024 · abfc6c2 · abfc6c2
1 parent 7c3b10f
commit abfc6c2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/.github/actions/e2e/action.yml b/.github/actions/e2e/action.yml
@@ -21,11 +21,6 @@ runs:
       shell: bash
       env:
         MPLLOCALFREETYPE: "1"
-    # - name: Setup Graphviz
-    #   uses: ts-graphviz/setup-graphviz@v1
-    #   with:
-    #     ubuntu-skip-apt-update: true
-    #     macos-skip-brew-update: true
 
     - name: Run tests
       working-directory: ${{ inputs.working_directory }}

diff --git a/tests/test_e2e.py b/tests/test_e2e.py
@@ -6,6 +6,7 @@
 except ImportError:
     from neptune.new import Run, init_run
 
+import matplotlib as mpl
 import pytest
 from sklearn.cluster import KMeans
 from sklearn.dummy import (
@@ -16,6 +17,8 @@
 
 import neptune_sklearn as npt_utils
 
+mpl.use("agg")
+
 
 def test_classifier_summary(iris):
     with init_run() as run: