1212# See the License for the specific language governing permissions and
1313# limitations under the License.
1414
15- import re
1615import argparse
16+ import re
1717from pathlib import Path
1818
1919try :
2020 from semver import Version
2121except ImportError as exc :
22- raise ImportError ("Unable to import semver. Install semver by running `pip install semver`" ) from exc
22+ raise ImportError (
23+ "Unable to import semver. Install semver by running `pip install semver`"
24+ ) from exc
2325
2426DEV_PRERELEASE_TAG_PREFIX = "dev"
2527DEV_PRERELEASE_TAG_START = "dev0"
@@ -45,10 +47,14 @@ def extract_version(repo_root_path: Path) -> Version:
4547 if line .startswith ("__version__" ):
4648 if (m := rgx_ver .match (line .strip ())) is not None :
4749 if not m .groups ():
48- raise ValueError (f"Unable to find valid semver for __version__. Got: '{ line } )
50+ raise ValueError (
51+ f"Unable to find valid semver for __version__. Got: '{ line }
52+ )
4953 parsed_semver = m .group (1 )
5054 if not Version .is_valid (parsed_semver ):
51- raise ValueError (f"Invalid semver for __version__. Got: '{ parsed_semver } { line } )
55+ raise ValueError (
56+ f"Invalid semver for __version__. Got: '{ parsed_semver } { line }
57+ )
5258 return Version .parse (parsed_semver )
5359 raise ValueError (f"Unable to find valid semver for __version__. Got: '{ line } )
5460 raise ValueError ("Cannot parse version" )
@@ -67,10 +73,7 @@ def update_prerelease_version(repo_root_path: Path, version: Version):
6773 raise FileNotFoundError (f"Unable to find version file at location { version_file_path } )
6874
6975 with version_file_path .open () as f :
70- lines = [
71- rgx_ver .sub (f"__version__ = \" { str (version )} \" " , line )
72- for line in f
73- ]
76+ lines = [rgx_ver .sub (f'__version__ = "{ str (version )} , line ) for line in f ]
7477
7578 with version_file_path .open ("w" ) as f :
7679 f .write ("" .join (lines ))
@@ -98,8 +101,14 @@ def update_prerelease_version(repo_root_path: Path, version: Version):
98101 # If a PR is of a higher version AND the prerelease tag is reset, then do nothing
99102 # This captures the case during release where we might bump the release version
100103 # within a PR and reset tag back to dev0
101- if pr_version > master_version and pr_version .prerelease and pr_version .prerelease == DEV_PRERELEASE_TAG_START :
102- print ("This Pull Request is upgrading the package version to next release ... skipping bumping!" )
104+ if (
105+ pr_version > master_version
106+ and pr_version .prerelease
107+ and pr_version .prerelease == DEV_PRERELEASE_TAG_START
108+ ):
109+ print (
110+ "This Pull Request is upgrading the package version to next release ... skipping bumping!"
111+ )
103112 print ("If this is happening in error, please report it to the PennyLane team!" )
104113 elif pr_version .prerelease and pr_version .prerelease .startswith (DEV_PRERELEASE_TAG_PREFIX ):
105114 # If master branch does not have a prerelease (for any reason) OR does not have an ending number
@@ -109,7 +118,10 @@ def update_prerelease_version(repo_root_path: Path, version: Version):
109118 else :
110119 # If master branch does not have a prerelease (for any reason) OR does not have an ending number
111120 # Then default to the starting tag
112- if not master_version .prerelease or master_version .prerelease == DEV_PRERELEASE_TAG_PREFIX :
121+ if (
122+ not master_version .prerelease
123+ or master_version .prerelease == DEV_PRERELEASE_TAG_PREFIX
124+ ):
113125 next_prerelease_version = DEV_PRERELEASE_TAG_START
114126 else :
115127 # Generate the next prerelease version (eg: dev1 -> dev2). Sourcing from master version.
0 commit comments