Localize doctest cupy dependency from global to gpu.rst

chainer · Jan 15, 2019 · 34f5647 · 34f5647
1 parent e8bc196
commit 34f5647
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -347,7 +347,6 @@
 doctest_global_setup = '''
 import os
 import numpy as np
-import cupy
 import chainer
 from chainer.backends import cuda
 from chainer import Function, gradient_check, training, utils, Variable

diff --git a/docs/source/guides/gpu.rst b/docs/source/guides/gpu.rst
@@ -19,6 +19,10 @@ After reading this section, you will be able to:
 * Write model-parallel computing in Chainer
 * Write data-parallel computing in Chainer
 
+.. testsetup::
+
+   import cupy
+
 .. testcode::
    :hide: