Update notebook modifications for dev docs

agriyakhetarpal · agriyakhetarpal · commit ddacd76fd0fd · 2025-03-27T02:33:34.000+05:30
diff --git a/doc/conf.py b/doc/conf.py
@@ -682,6 +682,22 @@ def notebook_modification_function(notebook_content, notebook_filename):
     # imports inside functions
     code_lines.extend(["import matplotlib", "import pandas"])
 
+    # Work around https://github.com/jupyterlite/pyodide-kernel/issues/166
+    # and https://github.com/pyodide/micropip/issues/223 by installing the
+    # dependencies first, and then scikit-learn from Anaconda.org.
+    if "dev" in release:
+        dev_docs_specific_code = [
+            "import micropip",
+            "await micropip.install(['joblib', 'threadpoolctl', 'scipy'])",
+            "await micropip.install(\n"
+            "   'scikit-learn',\n"
+            "   index_urls='https://pypi.anaconda.org/scientific-python-nightly-wheels/simple',\n"
+            "   pre=True,\n"
+            ")",
+        ]
+
+        code_lines.extend(dev_docs_specific_code)
+
     if code_lines:
         code_lines = ["# JupyterLite-specific code"] + code_lines
         code = "\n".join(code_lines)
