Updated the keras_autodoc version. (keras-team#125)

ucdmkt · Oct 22, 2019 · f02ed62 · f02ed62
1 parent 7f6b00f
commit f02ed62
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -36,7 +36,7 @@ install:
   - travis_retry conda install portpicker
 
   # install tools for building the docs
-  - pip install mkdocs git+https://github.com/keras-team/keras-autodoc.git@v0.1.3
+  - pip install mkdocs git+https://github.com/keras-team/keras-autodoc.git@v0.2.0
 
   - pip install -e .[tests] --progress-bar off
 

diff --git a/kerastuner_docs/autogen.py b/kerastuner_docs/autogen.py
@@ -48,13 +48,15 @@
 
 def generate(dest_dir):
     template_dir = kerastuner_dir / 'kerastuner_docs' / 'templates'
-    keras_autodoc.generate(
-        dest_dir,
-        template_dir,
+
+    doc_generator = keras_autodoc.DocumentationGenerator(
         PAGES,
         'https://github.com/keras-team/keras-tuner/blob/master',
-        kerastuner_dir / 'examples',
+        template_dir,
+        kerastuner_dir / 'examples'
     )
+    doc_generator.generate(dest_dir)
+
     readme = (kerastuner_dir / 'README.md').read_text()
     index = (template_dir / 'index.md').read_text()
     index = index.replace('{{autogenerated}}', readme[readme.find('##'):])