Read TF version from poetry.lock

.github/workflows/ci-model-regression-on-schedule.yml

@@ -59,10 +59,9 @@ jobs:
 
       - name: Get TensorFlow version
         run: |-
-          # Read TF version, trim special characters ^~><+
-          TF_VERSION=$(grep "tensorflow =" pyproject.toml | sed 's/^[^"]*"\([^"]*\),.*/\1/' | sed 's/[\^~><=]//g')
-          # Keep the first 3 characters, e.g. we keep 2.3 if TF_VERSION is 2.3.4
-          TF_VERSION=${TF_VERSION::3}
+          # Read TF version from poetry.lock file
+          pip install toml
+          TF_VERSION=$(scripts/read_tenforflow_version.sh)
           echo "TensorFlow version: $TF_VERSION"
           echo TF_VERSION=$TF_VERSION >> $GITHUB_ENV
 

.github/workflows/ci-model-regression.yml

@@ -155,8 +155,9 @@ jobs:
 
       - name: Get TensorFlow version
         run: |-
-          # Read TF version, trim special characters ^~><+
-          TF_VERSION=$(grep "tensorflow =" pyproject.toml | sed 's/^[^"]*"\([^"]*\),.*/\1/' | sed 's/[\^~><=]//g')
+          # Read TF version from poetry.lock file
+          pip install toml
+          TF_VERSION=$(scripts/read_tenforflow_version.sh)
           # Keep the first 3 characters, e.g. we keep 2.3 if TF_VERSION is 2.3.4
           TF_VERSION=${TF_VERSION::3}
           echo "TensorFlow version: $TF_VERSION"

scripts/read_tenforflow_version.sh

@@ -0,0 +1,22 @@
+#!/usr/bin/env python
+
+"""Extract the TensorFlow version from poetry.lock. Used in e.g. github workflows."""
+
+import pathlib
+import sys
+import toml
+
+poetry_path = 'poetry.lock'
+
+if __name__ == "__main__":
+    if pathlib.Path(poetry_path).is_file():
+      # Get the list of dict that contains all information about packages
+      package_list = toml.load(poetry_path).get('package')
+      # Extract tensorflow version
+      tf_version = next(filter(lambda x: x.get('name') == 'tensorflow', package_list)).get('version')
+      print(tf_version)
+      sys.exit(0)
+
+    else:
+      print(f'File {poetry_path} does not exist.')
+      sys.exit(1)