Apply suggestions from code review

tomschr · tlaferriere · web-flow · commit 14e4f96df8c5 · 2020-11-20T21:47:43.000+01:00
Co-authored-by: Thomas Laferriere &lt;t.laferriere@hotmail.ca&gt;
diff --git a/src/semver/version.py b/src/semver/version.py
@@ -182,8 +182,14 @@ def _check_types(*args):
                     "{!r} is negative. A version can only be positive.".format(name)
                 )
 
-        prerelease = cls._ensure_str(prerelease or verlist[3])  # type: ignore
-        build = cls._ensure_str(build or verlist[4])  # type: ignore
+        if isinstance(prerelease, int):
+            self._prerelease = prerelease
+        else:
+            self._prerelease = cls._ensure_str(prerelease or verlist[3])
+        if isinstance(build, int):
+            self._build= build
+        else:
+            self._build = cls._ensure_str(build or verlist[4])
 
         self._major = version_parts["major"]
         self._minor = version_parts["minor"]
@@ -216,8 +222,8 @@ def cmp_prerelease_tag(a, b):
 
     @classmethod
     def _ensure_str(
-        cls, s: Optional[StringOrInt], encoding="UTF-8"
-    ) -> Optional[StringOrInt]:
+        cls, s: Optional[String], encoding="UTF-8"
+    ) -> Optional[str]:
         """
         Ensures string type regardless if argument type is str or bytes.
 
