From 0d32c0c8d39b2415b7cad61033b94a6fcef0bb42 Mon Sep 17 00:00:00 2001
From: nicor88 <6278547+nicor88@users.noreply.github.com>
Date: Mon, 15 Apr 2024 21:17:24 +0200
Subject: [PATCH] fix: allow release to be beta (#624)

---
 setup.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index afe69bc3..f19cbd83 100644
--- a/setup.py
+++ b/setup.py
@@ -28,7 +28,10 @@ def _get_plugin_version_dict() -> Dict[str, Any]:
 
 def _get_package_version() -> str:
     parts = _get_plugin_version_dict()
-    return f'{parts["major"]}.{parts["minor"]}.{parts["patch"]}'
+    version = "{major}.{minor}.{patch}".format(**parts)
+    if parts["prekind"] and parts["pre"]:
+        version += parts["prekind"] + parts["pre"]
+    return version
 
 
 description = "The athena adapter plugin for dbt (data build tool)"