fixup version from tag for supporting beta

dschep · Apr 15, 2016 · 377fb9c · 377fb9c
1 parent dab8f22
commit 377fb9c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -18,8 +18,8 @@
     version = None
 else:
     version_parts = version_output.decode().strip().lstrip('v').split('-')
-    if len(version_parts) == 1:
-        version = version_parts[0]
+    if len(version_parts) <= 2:
+        version = '-'.join(version_parts)
     elif len(version_parts) > 1:
         version = '-'.join(version_parts[:-1])