fix(rtd): Correct doc rendering.

vinci1it2000 · Nov 3, 2022 · 82c9924 · 82c9924
1 parent 2021156
commit 82c9924
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/schedula/dispatcher.py b/schedula/dispatcher.py
@@ -1055,7 +1055,8 @@ def set_default_value(self, data_id, value=EMPTY, initial_dist=0.0):
 
         A dispatcher with a data node named `a`::
 
-            >>> dsp = Dispatcher(name='Dispatcher')
+            >>> import schedula as sh
+            >>> dsp = sh.Dispatcher(name='Dispatcher')
             ...
             >>> dsp.add_data(data_id='a')
             'a'
@@ -1068,7 +1069,7 @@ def set_default_value(self, data_id, value=EMPTY, initial_dist=0.0):
 
         Remove the default value of `a` node::
 
-            >>> dsp.set_default_value('a', value=EMPTY)
+            >>> dsp.set_default_value('a', value=sh.EMPTY)
             >>> dsp.default_values
             {}
         """

diff --git a/schedula/ext/dispatcher/documenter.py b/schedula/ext/dispatcher/documenter.py
@@ -144,7 +144,7 @@ def _plot(lines, dsp, dot_view_opt, documenter, dsp_opt):
     executor = dot_view_opt.get('executor', 'async')
     if osp.isdir(dspdir):
         fpath = glob.glob(osp.join(dspdir, '*.gv'))[0]
-    else:
+    elif hasattr(dsp, 'plot'):
         smap = dsp.plot(**dot_view_opt)
         folder = next(iter(smap))
         fpath = osp.join(dspdir, '%s.gv' % folder.sitemap.foldername)
@@ -153,6 +153,8 @@ def _plot(lines, dsp, dot_view_opt, documenter, dsp_opt):
             directory=dspdir, index=index, viz_js=viz_js, executor=executor
         )
         dot.save(fpath, directory='')
+    else:
+        return
 
     dsource = osp.dirname(osp.join(env.srcdir, env.docname))
     p = osp.relpath(fpath, dsource).replace('\\', '/')