keep both version of toc JS script, and use a better name

docs/_static/build_toc.js

+// modified from
+// https://stackoverflow.com/questions/12150491/toc-list-with-all-classes-generated-by-automodule-in-sphinx-docs
+
+$(function (){
+  var createList = function(selected) {
+    var ul = $('<ul>');
+
+    selected.each(function(i,e) {
+
+      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>');
+      a.attr('href',l.attr('href')).attr('title', 'Link to this definition');
+
+      a.append(n);
+
+      var entry = $('<li>').append(a);
+      ul.append(entry);
+    });
+    return ul;
+  }
+
+
+
+  var customIndex = $('.custom-index');
+  customIndex.empty();
+
+  var selected = $('div.section>dl>dt');
+  if (selected.length === 0)
+    return;
+
+  var l = createList(selected);
+
+  var c = $('<div style="min-width: 300px;">');
+  var ul = c.clone()
+    .append(l);
+  customIndex.append(ul);
+
+});

docs/modules/callbacks.rst

@@ -5,7 +5,7 @@ tensorpack.callbacks package
 
     .. raw:: html
 
-        <script type="text/javascript" src='../_static/pymunk.js'></script>
+        <script type="text/javascript" src='../_static/build_toc_group.js'></script>
 
 
 .. automodule:: tensorpack.callbacks

docs/modules/dataflow.dataset.rst

@@ -5,7 +5,7 @@ tensorpack.dataflow.dataset package
 
     .. raw:: html
 
-        <script type="text/javascript" src='../_static/pymunk.js'></script>
+        <script type="text/javascript" src='../_static/build_toc.js'></script>
 
 
 .. automodule:: tensorpack.dataflow.dataset

docs/modules/dataflow.imgaug.rst

@@ -5,7 +5,7 @@ tensorpack.dataflow.imgaug package
 
     .. raw:: html
 
-        <script type="text/javascript" src='../_static/pymunk.js'></script>
+        <script type="text/javascript" src='../_static/build_toc_group.js'></script>
 
 .. automodule:: tensorpack.dataflow.imgaug
     :members:

docs/modules/dataflow.rst

 tensorpack.dataflow package
 ===========================
 
-Table of Contents
------------------
 .. toctree::
 
     dataflow.dataset
@@ -12,7 +10,7 @@ Table of Contents
 
     .. raw:: html
 
-        <script type="text/javascript" src='../_static/pymunk.js'></script>
+        <script type="text/javascript" src='../_static/build_toc_group.js'></script>
 
 
 DataFlow

docs/modules/models.rst

@@ -5,7 +5,7 @@ tensorpack.models package
 
     .. raw:: html
 
-        <script type="text/javascript" src='../_static/pymunk.js'></script>
+        <script type="text/javascript" src='../_static/build_toc.js'></script>
 
 
 .. automodule:: tensorpack.models

docs/modules/train.rst

@@ -5,7 +5,7 @@ tensorpack.train package
 
     .. raw:: html
 
-        <script type="text/javascript" src='../_static/pymunk.js'></script>
+        <script type="text/javascript" src='../_static/build_toc.js'></script>
 
 .. automodule:: tensorpack.train
     :members: