diff --git a/scripts/update_version.py b/scripts/update_version.py index fb9b15dd31..003edee274 100644 --- a/scripts/update_version.py +++ b/scripts/update_version.py @@ -30,6 +30,16 @@ def update_version(): file.seek(0) file.write(replace_version) + # pyproject.toml + with open(os.path.join(pybamm.root_dir(), "pyproject.toml"), "r+") as file: + output = file.read() + replace_version = re.sub( + '(?<=version = ")(.+)(?=")', release_version, output + ) + file.truncate(0) + file.seek(0) + file.write(replace_version) + # CITATION.cff with open(os.path.join(pybamm.root_dir(), "CITATION.cff"), "r+") as file: output = file.read()