Move imgaug and dataset to top-level doc tree

docs/_static/build_toc.js

@@ -6,17 +6,16 @@ $(function (){
     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);
+
+      var n = $(e).children('.descname').clone();
       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);

docs/_static/build_toc_group.js

@@ -7,39 +7,35 @@ $(function (){
     var ul = $('<ul>');
 
     selected.each(function(i,e) {
-      
-      var className = e.getElementsByTagName('a')[0].href;
-      className = className.substr(className.lastIndexOf('/')+1);
-      className = className.substr(0,className.lastIndexOf(".html"));
 
+      var groupName = e.getElementsByTagName('a')[0].href;
+      groupName = groupName.substr(groupName.lastIndexOf('/')+1);
+      groupName = groupName.substr(0,groupName.lastIndexOf(".html"));
 
 
       var fullname = e.id;
       if (fullname.startsWith('tensorpack.'))
         fullname = fullname.substr(11);
+
       var n = $(e).children('.descname').clone();
-      
       n[0].innerText = fullname;
-      var l = $(e).children('.headerlink');
 
+      var l = $(e).children('.headerlink');
       var a = $('<a>');
-      a.attr('href',l.attr('href')).attr('title', 'Link to this definition');
-
+      a.attr('href', l.attr('href')).attr('title', 'Link to this definition');
       a.append(n);
 
       var entry = $('<li>').append(a);
 
-      if(className in obj) {
-        obj[className] = obj[className].append(entry);
+      if(groupName in obj) {
+        obj[groupName].append(entry);
       } else {
         var ul = $('<ul>');
         ul.append(entry);
-        obj[className] = ul;
+        obj[groupName] = ul;
       }
-      
     });
-    
-    
+
     return obj;
   }
 
@@ -48,20 +44,18 @@ $(function (){
   var customIndex = $('.custom-index');
   customIndex.empty();
 
-  
+
   var selected = $('div.section>dl>dt');
   if (selected.length === 0)
     return;
 
   var obj = createList(selected);
+  var block = $('<div style="min-width: 300px;">');
   for(var key in obj) {
-    var c = $('<div style="min-width: 300px;">');
-    var a = $('<h4>');
-    a.html(key);
-    var u = c.clone().append(a);
-    var ul = c.clone().append(obj[key]);
-    customIndex.append(u);
-    customIndex.append(ul);
+    var a = $('<h5>');
+    a.html(key + ':');
+    block.append(a);
+    block.append(obj[key]);
   }
-
+  customIndex.append(block);
 });

docs/modules/dataflow.rst

 tensorpack.dataflow package
 ===========================
 
-.. toctree::
-
-    dataflow.dataset
-    dataflow.imgaug
-
 .. container:: custom-index
 
     .. raw:: html

docs/modules/index.rst

@@ -6,6 +6,8 @@ API Documentation
 
 
   dataflow
+  dataflow.dataset
+  dataflow.imgaug
   models
   callbacks
   graph_builder

docs/modules/tfutils.rst

 tensorpack.tfutils package
 ==========================
 
-Module contents
----------------
-
-.. automodule:: tensorpack.tfutils
-    :members:
-    :undoc-members:
-    :show-inheritance:
-
 tensorpack.tfutils.collection module
 ------------------------------------
 
@@ -96,3 +88,12 @@ tensorpack.tfutils.varreplace module
     :members:
     :undoc-members:
     :show-inheritance:
+
+Module contents
+---------------
+
+.. automodule:: tensorpack.tfutils
+    :members:
+    :undoc-members:
+    :show-inheritance:
+