Skip to content

Commit 28dee22

Browse files
committed
get Sphinx documentation to build all optimizers
1 parent 6a06766 commit 28dee22

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

deepxde/optimizers/__init__.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,17 @@
55
from ..backend import backend_name
66

77

8+
# To get Sphinx documentation to build, we import all
9+
if os.environ.get("READTHEDOCS") == "True":
10+
# The backend should be tensorflow/tensorflow.compat.v1 to ensure backend.tf is not
11+
# None.
12+
from . import jax
13+
from . import paddle
14+
from . import pytorch
15+
from . import tensorflow
16+
from . import tensorflow_compat_v1
17+
18+
819
def _load_backend(mod_name):
920
mod = importlib.import_module(".%s" % mod_name, __name__)
1021
thismod = sys.modules[__name__]

0 commit comments

Comments
 (0)