added mock modules for matplotlib.cm and .tri
@@ -82,6 +82,7 @@ import mock
MOCK_MODULES = [
'numpy', 'scipy', 'pandas', 'seaborn',
'matplotlib', 'matplotlib.pyplot', 'matplotlib.animation', 'matplotlib.colors',
+ 'matplotlib.tri', 'matplotlib.cm',
'torch', 'torch.autograd', 'torch.nn', 'torch.optim'
]
for mod_name in MOCK_MODULES:
