Merge pull request #976 from vilyaair/Agg

Avoid to import matplotlib to set its backend Agg in code like chainer

examples/gmm/README.md

@@ -0,0 +1,19 @@
+# GMM example
+
+This example contains implementation of Gaussian Mixture Model (GMM).
+
+
+### How to demo
+The demo contains a script that partitions data into groups using Gaussian Mixture Model.
+The demo can be run by the following command.
+
+```
+python gmm.py [--gpu-id GPU_ID] [--num NUM] [--dim DIM]
+              [--max-iter MAX_ITER] [--tol TOL] [--output-image OUTPUT]
+```
+
+If you run this script on environment without matplotlib renderers (e.g., non-GUI environment), setting the environmental variable `MPLBACKEND` to `Agg` may be required to use `matplotlib`. For example,
+
+```
+MPLBACKEND=Agg python gmm.py ...
+```

examples/gmm/gmm.py

@@ -2,8 +2,6 @@
 import contextlib
 import time
 
-import matplotlib
-matplotlib.use('Agg')
 from matplotlib import mlab
 import matplotlib.pyplot as plt
 import numpy as np

examples/kmeans/README.md

@@ -0,0 +1,20 @@
+# kmeans example
+
+This example contains implementation of K-means clustering.
+
+
+### How to demo
+The demo contains a script that partitions data into groups using K-means clustering.
+The demo can be run by the following command.
+
+```
+python kmeans.py [--gpu-id GPU_ID] [--n-clusters N_CLUSTERS] [--num NUM]
+                 [--max-iter MAX_ITER] [--use-custom-kernel]
+                 [--output-image OUTPUT_IMAGE]
+```
+
+If you run this script on environment without matplotlib renderers (e.g., non-GUI environment), setting the environmental variable `MPLBACKEND` to `Agg` may be required to use `matplotlib`. For example,
+
+```
+MPLBACKEND=Agg python kmeans.py ...
+```

examples/kmeans/kmeans.py

@@ -2,8 +2,6 @@
 import contextlib
 import time
 
-import matplotlib
-matplotlib.use('Agg')
 import matplotlib.pyplot as plt
 import numpy as np
 import six

tests/example_tests/test_gmm.py

@@ -10,6 +10,9 @@
 from example_tests import example_test
 
 
+os.environ['MPLBACKEND'] = 'Agg'
+
+
 @testing.with_requires('matplotlib')
 class TestGMM(unittest.TestCase):
 

tests/example_tests/test_kmeans.py

@@ -10,6 +10,9 @@
 from example_tests import example_test
 
 
+os.environ['MPLBACKEND'] = 'Agg'
+
+
 @testing.with_requires('matplotlib')
 class TestKmeans(unittest.TestCase):