Add TOC to more documentation pages

docs/_static/pymunk.js

@@ -5,9 +5,13 @@ $(function (){
   var createList = function(selected) {
     var ul = $('<ul>');
 
-    selected.clone().each(function(i,e) {
+    selected.each(function(i,e) {
 
-      var n = $(e).children('.descname');
+      var fullname = e.id;
+      var n = $(e).children('.descname').clone();
+      if (fullname.startsWith('tensorpack.'))
+        fullname = fullname.substr(11);
+      n[0].innerText = fullname;
       var l = $(e).children('.headerlink');
 
       var a = $('<a>');
@@ -26,12 +30,7 @@ $(function (){
   var customIndex = $('.custom-index');
   customIndex.empty();
 
-  var mod_content = $('#module-contents');
-  if (mod_content.length == 0)
-    return;
-  var mc_container = mod_content[0].parentNode;
-  var selected = $(mc_container).find('>dl>dt');
-
+  var selected = $('div.section>dl>dt');
   if (selected.length === 0)
     return;
 

docs/_templates/layout.html

+{% extends "!layout.html" %}
+
+{%- block extrahead %}
+    <script type="text/javascript" src='/_static/jquery-3.2.1.min.js'></script>
+{% endblock %}

docs/conf.py

@@ -160,7 +160,7 @@ html_theme_path = [sphinx_rtd_theme.get_html_theme_path()]
 # Theme options are theme-specific and customize the look and feel of a theme
 # further.  For a list of options available for each theme, see the
 # documentation.
-html_theme_options = {}
+html_theme_options = { }
 
 # The name for this set of Sphinx documents.  If None, it defaults to
 # "<project> v<release> documentation".

docs/modules/callbacks.rst

 tensorpack.callbacks package
 ============================
 
+.. container:: custom-index
+
+    .. raw:: html
+
+        <script type="text/javascript" src='/_static/pymunk.js'></script>
+
+
 .. automodule:: tensorpack.callbacks
     :members:
     :no-undoc-members:

docs/modules/dataflow.dataset.rst

 tensorpack.dataflow.dataset package
 ===================================
 
+.. container:: custom-index
+
+    .. raw:: html
+
+        <script type="text/javascript" src='/_static/pymunk.js'></script>
+
+
 .. automodule:: tensorpack.dataflow.dataset
     :members:
     :undoc-members:

docs/modules/dataflow.imgaug.rst

 tensorpack.dataflow.imgaug package
 ==================================
 
+.. container:: custom-index
+
+    .. raw:: html
+
+        <script type="text/javascript" src='/_static/pymunk.js'></script>
+
 .. automodule:: tensorpack.dataflow.imgaug
     :members:
     :undoc-members:

docs/modules/dataflow.rst

 tensorpack.dataflow package
 ===========================
 
-Subpackages
------------
-
+Table of Contents
+-----------------
 .. toctree::
 
     dataflow.dataset
     dataflow.imgaug
 
-Table of Contents
------------------
 .. container:: custom-index
 
     .. raw:: html
 
-        <script src="https://ajax.googleapis.com/ajax/libs/jquery/3.2.1/jquery.min.js"></script>
         <script type="text/javascript" src='/_static/pymunk.js'></script>
 
 
-Module Contents
+DataFlow
 ---------------
 
 .. automodule:: tensorpack.dataflow

docs/modules/models.rst

 tensorpack.models package
 =========================
 
+.. container:: custom-index
+
+    .. raw:: html
+
+        <script type="text/javascript" src='/_static/pymunk.js'></script>
+
+
 .. automodule:: tensorpack.models
     :members:
     :undoc-members:

tensorpack/models/__init__.py

@@ -8,7 +8,7 @@ import os.path
 # this line is necessary for _TFModuleFunc to work
 import tensorflow as tf  # noqa: F401
 
-__all__ = ['LinearWrap']
+__all__ = []
 
 
 def _global_import(name):