🧮 Require location on node source to enable jupyter computation (#381)

jupyter-book · May 9, 2024 · 21f72bd · 21f72bd
1 parent 0cea444
commit 21f72bd
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 1 deletion.
diff --git a/.changeset/giant-bugs-rest.md b/.changeset/giant-bugs-rest.md
@@ -0,0 +1,5 @@
+---
+'@myst-theme/jupyter': patch
+---
+
+Require location on node source to enable jupyter computation
diff --git a/packages/jupyter/src/decoration.tsx b/packages/jupyter/src/decoration.tsx
@@ -32,20 +32,25 @@ export function OutputDecoration({
   children,
   title = 'Jupyter Notebook',
   url,
+  location,
 }: {
   outputId: string;
   placeholder?: GenericNode;
   children?: React.ReactNode;
   title?: string;
   url?: string;
+  location?: string;
 }) {
   const { kind } = useCellExecution(outputId);
   const compute = useComputeOptions();
   const Link = useLinkProvider();
   const top = useThemeTop();
   const baseurl = useBaseurl();
   const showComputeControls =
-    compute?.enabled && compute?.features.figureCompute && kind === SourceFileKind.Article;
+    compute?.enabled &&
+    compute?.features.figureCompute &&
+    kind === SourceFileKind.Article &&
+    location;
 
   if (showComputeControls) {
     return (

diff --git a/packages/jupyter/src/embed.tsx b/packages/jupyter/src/embed.tsx
@@ -11,6 +11,7 @@ export function Embed({ node }: { node: GenericNode }) {
       outputId={output.id}
       title={node.source?.title}
       url={node.source?.url}
+      location={node.source?.location}
     >
       <MyST ast={node.children} />
     </OutputDecoration>

diff --git a/packages/jupyter/src/figure.tsx b/packages/jupyter/src/figure.tsx
@@ -22,6 +22,7 @@ export function Figure({ node }: { node: GenericNode }) {
           placeholder={placeholder}
           title={node.source?.title}
           url={node.source?.url}
+          location={node.source?.location}
         >
           <MyST ast={others} />
         </OutputDecoration>