Set backend in swerve matplotlib users
On some systems the default backend doesn't work. Replicate the setup
from build_tests/matplotlib_example.py.
Change-Id: I28cd8e4d3fc442a9aea0dcc26490ec4b5ca7ed83
Signed-off-by: James Kuszmaul <jabukuszmaul+collab@gmail.com>
diff --git a/frc971/control_loops/swerve/casadi_velocity_mpc.py b/frc971/control_loops/swerve/casadi_velocity_mpc.py
index f9981c9..698cce7 100644
--- a/frc971/control_loops/swerve/casadi_velocity_mpc.py
+++ b/frc971/control_loops/swerve/casadi_velocity_mpc.py
@@ -2,6 +2,7 @@
from frc971.control_loops.swerve import dynamics
import matplotlib.pyplot as pyplot
+import matplotlib
from matplotlib import pylab
import numpy
import time
@@ -9,6 +10,8 @@
import casadi
import os, sys
+matplotlib.use("GTK3Agg")
+
class MPC(object):
diff --git a/frc971/control_loops/swerve/simulation.py b/frc971/control_loops/swerve/simulation.py
index 4eed97b..d7e2fec 100644
--- a/frc971/control_loops/swerve/simulation.py
+++ b/frc971/control_loops/swerve/simulation.py
@@ -6,6 +6,7 @@
from frc971.control_loops.python import control_loop
from frc971.control_loops.python import controls
+import matplotlib
from matplotlib import pylab
import sys
import gflags
@@ -13,6 +14,8 @@
FLAGS = gflags.FLAGS
+matplotlib.use("GTK3Agg")
+
class SwerveSimulation(object):