vela: Improve regex for readme link replacement

Instead of just detecting MD files, detect any local files by checking
if they exist

Signed-off-by: Michael McGeagh <michael.mcgeagh@arm.com>
Change-Id: I02bd790b5bb148abfe16e4bdb3e38e68750cfb94
diff --git a/RELEASES.md b/RELEASES.md
index ac10e8b..16cca1e 100644
--- a/RELEASES.md
+++ b/RELEASES.md
@@ -5,6 +5,10 @@
 fixed.  The version numbering adheres to the
 [semantic versioning](https://semver.org/) scheme.
 
+## Release 2.0.1 - 03/12/2020
+
+* Fixed a broken link in the long description used by PyPi
+
 ## Release 2.0.0 - 30/11/2020
 
 **Main feature changes:**
diff --git a/setup.py b/setup.py
index f5f82bd..ee4e40e 100644
--- a/setup.py
+++ b/setup.py
@@ -26,12 +26,12 @@
 this_directory = path.abspath(path.dirname(__file__))
 with open(path.join(this_directory, "README.md"), encoding="utf-8") as f:
     long_description = f.read()
-    # Replace local Markdown links with URLs
-    tag = "2.0.0"
+    # Replace local file links with URLs
+    tag = "2.0.1"
     url = f"https://review.mlplatform.org/plugins/gitiles/ml/ethos-u/ethos-u-vela/+/refs/tags/{tag}/"
-    for markdown in set(re.findall(r"\(.+\.md\)", long_description)):
-        link = f"({url}{markdown[1:-1]})"
-        long_description = long_description.replace(markdown, link)
+    for link in set(re.findall(r"\[.+\]\((.+?)\)", long_description)):
+        if path.exists(path.join(this_directory, link)):
+            long_description = long_description.replace(link, url + link)
 
 mlw_module = Extension(
     "ethosu.mlw_codec",