Refactor setup.py: Modify version tagging logic to temporarily adjust…

… version for acceleration
locaal-ai · Oct 30, 2024 · 4136426 · 4136426
1 parent f84c3cb
commit 4136426
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -105,8 +105,10 @@ def get_tag(self):
         python, abi, platform = super().get_tag()
         acceleration = os.environ.get("SIMPLER_WHISPER_ACCELERATION", "")
         if acceleration:
-            # This creates the +cuda or +cpu tag
-            self.distribution.version += f"+{acceleration}"
+            # Store original version
+            orig_version = self.distribution.get_version()
+            # Temporarily modify version
+            self.distribution.metadata.version = f"{orig_version}+{acceleration}"
         return python, abi, platform