add bmnist example

jax-ml · Apr 5, 2024 · 13bab04 · 13bab04
1 parent 3ce8c3d
commit 13bab04
Show file tree

Hide file tree

Showing 8 changed files with 174 additions and 138 deletions.
diff --git a/docs/_static/bmnist.gif b/docs/_static/bmnist.gif
diff --git a/docs/_static/dmm.png b/docs/_static/dmm.png
diff --git a/docs/_static/dmm_oryx.png b/docs/_static/dmm_oryx.png
diff --git a/docs/conf.py b/docs/conf.py
@@ -207,11 +207,13 @@
 ):
   toctree_path = "notebooks/" if src_file.endswith("ipynb") else "examples/"
   filename = os.path.splitext(src_file.split("/")[-1])[0]
-  png_path = "_static/" + filename + ".png"
+  img_path = "_static/" + filename + ".png"
   # use Coix logo if not exist png file
-  if not os.path.exists(png_path):
-    png_path = "_static/coix_logo.png"
-  nbsphinx_thumbnails[toctree_path + filename] = png_path
+  if not os.path.exists(img_path):
+    img_path = "_static/" + filename + ".gif"
+    if not os.path.exists(img_path):
+      img_path = "_static/coix_logo.png"
+  nbsphinx_thumbnails[toctree_path + filename] = img_path
 
 
 # -- Options for HTML output -------------------------------------------------