Merge branch 'main' into fix/ppo-mujoco-colab

Author: abhitorch81

.ci/docker/requirements.txt

@@ -2,7 +2,7 @@
 # Refer to ./jenkins/build.sh for tutorial build instructions
 
 sphinx==5.3.0
-sphinx-gallery==0.11.1
+sphinx-gallery==0.17.1
 sphinx-reredirects==0.1.4
 sphinx-design==0.4.0
 docutils==0.16
@@ -31,7 +31,8 @@ pytorch-lightning
 torchx
 torchrl==0.7.2
 tensordict==0.7.2
-ax-platform>=0.4.0
+# For ax_multiobjective_nas_tutorial.py
+ax-platform>=0.4.0,<0.5.0
 nbformat>=5.9.2
 datasets
 transformers

conf.py

@@ -85,7 +85,10 @@ def wrapper(*args, **kwargs):
             raise RuntimeError(f"Error in subprocess: {result}")
     return wrapper
 
-sphinx_gallery.gen_rst.generate_file_rst = call_in_subprocess(sphinx_gallery.gen_rst.generate_file_rst)
+# Windows does not support multiprocessing with fork and mac has issues with
+# fork so we do not monkey patch sphinx gallery to run in subprocesses.
+if os.getenv("TUTORIALS_ISOLATE_BUILD", "1") == "1" and not sys.platform.startswith("win") and not sys.platform == "darwin":
+    sphinx_gallery.gen_rst.generate_file_rst = call_in_subprocess(sphinx_gallery.gen_rst.generate_file_rst)
 
 try:
     import torchvision

custom_directives.py

@@ -88,7 +88,8 @@ def run(self):
             if 'intro' in self.options:
                 intro = self.options['intro'][:195] + '...'
             else:
-                _, blocks = sphinx_gallery.gen_rst.split_code_and_text_blocks(abs_fname)
+                block_parser = sphinx_gallery.gen_rst.BlockParser(abs_fname, {"filetype_parsers": {}})
+                _, blocks, _ = block_parser.split_code_and_text_blocks(abs_fname)
                 intro, _ = sphinx_gallery.gen_rst.extract_intro_and_title(abs_fname, blocks[0][1])
 
             thumbnail_rst = ''