remove suffix added by fork build process when setting ONNX required version

Author: Kiko-Aumond

setup.py

@@ -16,14 +16,15 @@
   # For user, we install the onnx release known to work with our release.
   with open(os.path.join(TOP_DIR, 'ONNX_VERSION_NUMBER')) as onnx_version_file:
     onnx_version = onnx_version_file.read().strip()
-    onnx_dep = "onnx>=" + onnx_version
-    # For ONNX 1.9.0 for the Bonsai fork we need to use ONNX 1.9.0
-    # since we don't have an internal ONNX fork and the onnx_version above
-    # will be the one generated by the build process (1.9.0.xxxx), which
-    # does not exist in the ONNX repo
-    if onnx_version.startswith("1.9.0"):
-        onnx_dep = "onnx==1.9.0"
+    # see if our build process has added a suffix to the version
+    n_dots = [i for i, letter in enumerate(s) if letter == "."]
 
+    # get the ONNX version from the upstream repo
+    # by removing the suffix appended by the Fork build process
+    if len(n_dots) == 3:
+        onnx_version = onnx_version[0:n_dots[-1]]
+
+    onnx_dep = "onnx==" + onnx_version
 try:
   git_version = subprocess.check_output(['git', 'rev-parse', 'HEAD'],
                                         cwd=TOP_DIR).decode('ascii').strip()