fix version

magmax · Feb 28, 2021 · 00fec25 · 00fec25
1 parent 33faa81
commit 00fec25
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -1,12 +1,16 @@
 # -*- coding: utf-8 -*-
 
+import os
 import sys
 from io import open
 
 from setuptools import find_packages, setup
 from setuptools.command.test import test as TestCommand
 
 version = "2.0.1"
+github_ref = os.getenv("GITHUB_REF")
+if github_ref and github_ref.startswith("refs/tags"):
+    version = github_ref[10:]
 
 
 def read_description():