Remove matplotlib in minimal test setting

SMTorg · Nov 24, 2023 · 21d1b52 · 21d1b52
1 parent a33b5e7
commit 21d1b52
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/tests_minimal.yml b/.github/workflows/tests_minimal.yml
@@ -19,8 +19,7 @@ jobs:
     - name: Install dependencies
       run: |
         pip install --upgrade pip
-        pip install numpy cython 
-        pip install matplotlib   # required by GENN atm, to be fixed
+        pip install numpy cython
         pip install -e .
 
     - name: Test with pytest

diff --git a/smt/utils/neural_net/model.py b/smt/utils/neural_net/model.py
@@ -7,7 +7,7 @@
 """
 
 import numpy as np
-import matplotlib.gridspec as gridspec
+
 from smt.utils.neural_net.data import random_mini_batches
 from smt.utils.neural_net.optimizer import Adam
 from smt.utils.neural_net.activation import Tanh, Linear
@@ -441,6 +441,7 @@ def gradient(self, X):
 
     def goodness_of_fit(self, X_test, Y_test, J_test=None, response=0, partial=0):
         import matplotlib.pyplot as plt
+        import matplotlib.gridspec as gridspec
 
         assert X_test.shape[1] == Y_test.shape[1]
         assert Y_test.shape[0] == Y_test.shape[0]