From 6caa75a08924571b1d861a556c7139ee87438425 Mon Sep 17 00:00:00 2001
From: Daniel King <43149077+dakinggg@users.noreply.github.com>
Date: Sat, 20 Apr 2024 11:16:37 -0700
Subject: [PATCH] Change main to a dev version (#1126)

---
 llmfoundry/__init__.py | 2 +-
 setup.py               | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/llmfoundry/__init__.py b/llmfoundry/__init__.py
index 54a55d6e97..29b6552f01 100644
--- a/llmfoundry/__init__.py
+++ b/llmfoundry/__init__.py
@@ -62,4 +62,4 @@
     'registry',
 ]
 
-__version__ = '0.7.0'
+__version__ = '0.8.0.dev0'
diff --git a/setup.py b/setup.py
index b00d2be6fc..4bdb15aa18 100644
--- a/setup.py
+++ b/setup.py
@@ -21,7 +21,9 @@
     content = f.read()
 # regex: '__version__', whitespace?, '=', whitespace, quote, version, quote
 # we put parens around the version so that it becomes elem 1 of the match
-expr = re.compile(r"""^__version__\W+=\W+['"]([0-9\.]*)['"]""", re.MULTILINE)
+expr = re.compile(
+    r"""^__version__\s*=\s*['"]([0-9]+\.[0-9]+\.[0-9]+(?:\.\w+)?)['"]""",
+    re.MULTILINE)
 repo_version = expr.findall(content)[0]
 
 # Use repo README for PyPi description