fix master installation (#604)

* fix master installation * fix master installation * fix master installation * fix master installation * fix master installation
Deci-AI · Jan 8, 2023 · 3445523 · 3445523
1 parent 12ebdc2
commit 3445523
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -11,6 +11,7 @@
 REQ_LOCATION = "requirements.txt"
 REQ_PRO_LOCATION = "requirements.pro.txt"
 VERSION_FILE = "version.txt"
+INIT_FILE = "src/super_gradients/__init__.py"
 
 
 def readme():
@@ -32,7 +33,15 @@ def get_pro_requirements():
 
 def get_version():
     with open(VERSION_FILE, encoding="utf-8") as f:
-        return f.readline()
+        ver = f.readline()
+
+    if ver.startswith("for"):
+        with open(INIT_FILE, encoding="utf-8") as f:
+            for line in f.readlines():
+                if line.startswith("__version__"):
+                    ver = line.split()[-1].strip('"') + "+master"
+
+    return ver
 
 
 setup(