fix docs build (#141)

This fixes a few missing classes and a missing graph in the docs.
EmbarkStudios · Jun 8, 2023 · 104c0f7 · 104c0f7
1 parent 12fff94
commit 104c0f7
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
@@ -24,6 +24,8 @@ jobs:
     steps:
       - name: Checkout
         uses: actions/checkout@v3
+      - name: Setup Graphviz
+        uses: ts-graphviz/setup-graphviz@v1
       - name: Setup Pages
         uses: actions/configure-pages@v2
       - uses: pdm-project/setup-pdm@v3
@@ -37,7 +39,7 @@ jobs:
           cache: true
           cache-dependency-path: '**/pdm.lock'
       - name: Install dependencies
-        run: pdm install -d -G ci
+        run: pdm install -d -G ci && pdm torch install cpu
       - name: Build docs
         run: cd docs && make deploy
       - name: Upload artifact

diff --git a/docs/callbacks.rst b/docs/callbacks.rst
@@ -38,16 +38,16 @@ Checkpointing
 Logging callbacks and utilities
 -------------------------------
 
-.. currentmodule:: emote.callbacks.logging
-.. automodule:: emote.callbacks.logging
+.. currentmodule:: emote.callbacks
+.. automodule:: emote.callbacks
 
 .. autosummary::
     :toctree: generated
 
-    LoggingMixin
-	TensorboardLogger
-	TerminalLogger
-	WBLogger
+    logging.LoggingMixin
+	logging.TensorboardLogger
+	logging.TerminalLogger
+	wb_logger.WBLogger
 
 Testing
 -------