Make mpl optional in plots/traceplot. This way when modules import traceplot and try to resolve its imports, it doesn't throw ImportError.
+try:
import matplotlib.pyplot as plt
from matplotlib.ticker import MaxNLocator
+except ImportError: # mpl is optional
+ pass
import numpy as np
from keanu.vartypes import sample_types, numpy_types
