[UPDATE][MKDOCS]

kyegomez · Dec 27, 2023 · 430ac25 · 430ac25
1 parent 8dc0897
commit 430ac25
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/mkdocs.yml b/mkdocs.yml
@@ -129,6 +129,10 @@ nav:
  - QuickGELUActivation: "zeta/nn/modules/quickgeluactivation.md"
  - RecursiveBlock: "zeta/nn/modules/recursiveblock.md"
  - ReLUSquaredActivation: "zeta/nn/modules/relusquaredactivation.md"
+ - stochasticskipblock: "zeta/nn/modules/stochasticskipblock.md"
+ - gatedresidualblock: "zeta/nn/modules/gatedresidualblock.md"
+ - tripleskipblock: "zeta/nn/modules/tripleskipblock.md"
+ - DynamicRoutingBlock: "zeta/nn/modules/dynamicroutingblock.md"
  - zeta.nn.attention:
  - FlashAttention: "zeta/nn/attention/flash_attention.md"
  - MultiQueryAttention: "zeta/nn/attention/multiquery.md"

diff --git a/scripts/auto_tests_docs/mkdocs_handler.py b/scripts/auto_tests_docs/mkdocs_handler.py
@@ -22,8 +22,8 @@ def generate_file_list(directory, output_file):
  # Remove the file extension
  file_name, _ = os.path.splitext(file)
  # Write the file name and path to the output file
- f.write(f'- {file_name}: "{directory}{file_path}"\n')
+ f.write(f'- {file_name}: "{directory}/{file_path}"\n')
 
 
 # Use the function to generate the file list
-generate_file_list("docs/zeta/models", "file_list.txt")
+generate_file_list("docs/zeta/nn/modules", "file_list.txt")