Forcing TKAgg backend for matplotlib

david-abel · Apr 21, 2018 · e089dd9 · e089dd9
1 parent c6f1fec
commit e089dd9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/simple_rl/tasks/gather/GatherStateClass.py b/simple_rl/tasks/gather/GatherStateClass.py
@@ -4,7 +4,8 @@
 from simple_rl.mdp.StateClass import State
 import numpy as np
 import time
-
+import matplotlib
+matplotlib.use('TkAgg')   
 import matplotlib.pyplot as plt # NOTE: for debugging
 
 COLORS = {

diff --git a/simple_rl/utils/chart_utils.py b/simple_rl/utils/chart_utils.py
@@ -25,6 +25,7 @@
 import sys
 import os
 import matplotlib
+matplotlib.use('TkAgg')
 import matplotlib.pyplot as pyplot
 import numpy as np
 import subprocess