Fix issue with running tests on CI.

pypr · Dec 19, 2020 · e69bae0 · e69bae0
1 parent 61397fc
commit e69bae0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pysph/examples/dam_break_3d.py b/pysph/examples/dam_break_3d.py
@@ -95,8 +95,10 @@ def post_process(self, info_fname):
         if len(self.output_files) == 0:
             return
 
-        from pysph.solver.utils import iter_output
+        import matplotlib
+        matplotlib.use('Agg')
         import matplotlib.pyplot as plt
+        from pysph.solver.utils import iter_output
         from pysph.examples import db_exp_data as dbd
         from pysph.tools.interpolator import Interpolator
         H = self.geom.fluid_column_height