Skip to content

Commit bbd5a82

Browse files
committed
Remove center_badges function and update badges detail
1 parent 6f7171a commit bbd5a82

File tree

1 file changed

+6
-15
lines changed

1 file changed

+6
-15
lines changed

docs/update-readme-table.py

Lines changed: 6 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -8,18 +8,9 @@
88
with open("docs/notebooks-data.yml") as f:
99
data = yaml.safe_load(f)
1010

11-
12-
def center_badges(*badges):
13-
"""Center align badges, handling empty cells."""
14-
non_empty = [badge for badge in badges if badge.strip()]
15-
if not non_empty:
16-
return "<div align='center'>-</div>"
17-
return f"<div align='center'>{' '.join(non_empty)}</div>"
18-
19-
2011
# Generate table with proper alignment
2112
table = [
22-
"| Notebook | Open in colab / kaggle | Supporting materials | Discussion / arXiv / Repository |",
13+
"| Notebook | Open in colab / kaggle | Supporting materials | Documentation / arXiv / Repository |",
2314
"|:--------:|:----------------------:|:-------------------:|:-------------------------------:|",
2415
]
2516

@@ -44,7 +35,7 @@ def center_badges(*badges):
4435
else ""
4536
)
4637
dataset_badge = (
47-
f"[![Explore Dataset](https://github.com/user-attachments/assets/e9e2aec2-d8f7-492d-a65c-7ed09730a23d)]({nb['dataset']})"
38+
f"[![Explore Dataset](https://img.shields.io/badge/Dataset-%23ff1b6c?logo=ultralytics)]({nb['dataset']})"
4839
if nb.get("dataset")
4940
else ""
5041
)
@@ -59,7 +50,7 @@ def center_badges(*badges):
5950
else ""
6051
)
6152
discussion_badge = (
62-
f"[![Read Discussion](https://img.shields.io/badge/Documentation-E1FF25?logo=ultralytics)]({nb['discussion']})"
53+
f"[![Explore documentation](https://img.shields.io/badge/Documentation-111F68?logo=ultralytics)]({nb['discussion']})"
6354
if nb.get("discussion")
6455
else ""
6556
)
@@ -68,9 +59,9 @@ def center_badges(*badges):
6859
# Add row with proper centering
6960
table.append(
7061
f"| {title_link} | "
71-
f"{center_badges(colab_badge, kaggle_badge)} | "
72-
f"{center_badges(youtube_badge, dataset_badge, blog_badge)} | "
73-
f"{center_badges(arxiv_badge, discussion_badge, github_badge)} |"
62+
f"{colab_badge, kaggle_badge} | "
63+
f"{youtube_badge, dataset_badge, blog_badge} | "
64+
f"{arxiv_badge, discussion_badge, github_badge} |"
7465
)
7566

7667
table_md = "\n".join(table)

0 commit comments

Comments
 (0)