From 1c3fa1cc7aa6672e854967ac910b9d44d7564909 Mon Sep 17 00:00:00 2001
From: Maximilian Schulz <83698606+maxschulz-COL@users.noreply.github.com>
Date: Mon, 28 Oct 2024 18:16:04 +0100
Subject: [PATCH] [CI] PyCafe (#831)

---
 tools/pycafe/create_pycafe_links.py    | 6 ++++--
 vizro-core/examples/scratch_dev/app.py | 2 +-
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/tools/pycafe/create_pycafe_links.py b/tools/pycafe/create_pycafe_links.py
index e5d729363..fbd221059 100644
--- a/tools/pycafe/create_pycafe_links.py
+++ b/tools/pycafe/create_pycafe_links.py
@@ -10,14 +10,16 @@
 from typing import Optional
 from urllib.parse import quote, urlencode
 
+import vizro
 from github import Auth, Github
 
+PACKAGE_VERSION = vizro.__version__
+
 GITHUB_TOKEN = sys.argv[1]
 REPO_NAME = sys.argv[2]
 PR_NUMBER = int(sys.argv[3])
 RUN_ID = sys.argv[4]
 COMMIT_SHA = sys.argv[5]
-WHL_FILE = next(Path("dist").glob("*.whl")).name
 PYCAFE_URL = "https://py.cafe"
 VIZRO_RAW_URL = "https://raw.githubusercontent.com/mckinsey/vizro"
 
@@ -48,7 +50,7 @@ def generate_link(directory: str, extra_requirements: Optional[list[str]] = None
         filter(
             None,
             [
-                f"{PYCAFE_URL}/gh/artifact/mckinsey/vizro/actions/runs/{RUN_ID}/pip/{WHL_FILE}",
+                f"{PYCAFE_URL}/gh/artifact/mckinsey/vizro/actions/runs/{RUN_ID}/pip/vizro-{PACKAGE_VERSION}-py3-none-any.whl",
                 *(extra_requirements or []),
             ],
         )
diff --git a/vizro-core/examples/scratch_dev/app.py b/vizro-core/examples/scratch_dev/app.py
index 67306ecab..7bae4a92f 100644
--- a/vizro-core/examples/scratch_dev/app.py
+++ b/vizro-core/examples/scratch_dev/app.py
@@ -28,7 +28,7 @@
 
 
 page = vm.Page(
-    title="Test New II",
+    title="Test New III",
     components=[
         vm.Graph(
             figure=px.bar(