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: