fix: Demo page screenshot path (#76)

determined-ai · Dec 12, 2023 · e4fb70c · e4fb70c
1 parent d50a4c4
commit e4fb70c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/scripts/designkit-screenshot.mjs b/scripts/designkit-screenshot.mjs
@@ -28,7 +28,7 @@ const browser = await chromium.launch();
 const page = await browser.newPage();
 await page.goto(`http://${address}:${port}${publicUrl}/design/?exclusive=true`);
 // take screenshots of each section
-const links = await page.locator('nav ul a').all();
+const links = await page.locator('#main-nav ul a').all();
 if (links.length === 0) {
   console.error('WARNING: No sections found');
 }
@@ -40,7 +40,7 @@ for (const theme of THEMES) {
   for (const link of links) {
     await link.click();
     const title = await link.innerText();
-    const section = page.locator('div > article > article');
+    const section = page.locator('article');
     // playwright hangs if height is a non-int
     const height = Math.ceil((await section.boundingBox()).height);
     await page.setViewportSize({ height, width: 1280 });

diff --git a/src/DesignKit.tsx b/src/DesignKit.tsx
@@ -4588,7 +4588,7 @@ const DesignKit: React.FC<{
     <UIProvider priority="low" theme={theme} themeIsDark={themeIsDark}>
       <Spinner spinning={false}>
         <ElevationWrapper className={css.base} elevationOverride={0}>
-          <ElevationWrapper className={[css.nav, css.desktop].join(' ')}>
+          <ElevationWrapper className={[css.nav, css.desktop].join(' ')} id="main-nav">
             <ul className={css.sections}>
               <li>
                 <ThemeToggle mode={mode} onChange={onChangeMode} />