pypi azure badges - tags (#6068)

* pypi azure badges - tags * pep8 * id
Lightning-AI · Feb 18, 2021 · 3645eb1 · 3645eb1
1 parent e12c8a7
commit 3645eb1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pytorch_lightning/setup_tools.py b/pytorch_lightning/setup_tools.py
@@ -65,6 +65,9 @@ def _load_readme_description(path_dir: str, homepage: str = __homepage__, versio
     text = text.replace('/branch/master/graph/badge.svg', f'/release/{version}/graph/badge.svg')
     # replace github badges for release ones
     text = text.replace('badge.svg?branch=master&event=push', f'badge.svg?tag={version}')
+    # Azure...
+    text = text.replace('?branchName=master', f'?branchName=refs%2Ftags%2F{version}')
+    text = re.sub(r'\?definitionId=\d+&branchName=master', f'?definitionId=2&branchName=refs%2Ftags%2F{version}', text)
 
     skip_begin = r'<!-- following section will be skipped from PyPI description -->'
     skip_end = r'<!-- end skipping PyPI description -->'