Fix version parsing

gaogaotiantian · Aug 20, 2020 · e9ce2b1 · e9ce2b1
1 parent 7850b80
commit e9ce2b1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -7,7 +7,9 @@
 with open("./src/viztracer/__init__.py") as f:
     for line in f.readlines():
         if line.startswith("__version__"):
-            version = line.split("=")[-1].strip()
+            # __version__ = "0.9"
+            delim = '"' if '"' in line else "'"
+            version = line.split(delim)[1]
             break
     else:
         print("Can't find version! Stop Here!")