Building macOS images uses package version only

Signed-off-by: Enrico Minack <github@enrico.minack.dev>
horovod · Dec 8, 2021 · 96b01ec · 96b01ec
1 parent 4bc65ff
commit 96b01ec
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/gen-workflow-ci.py b/.github/gen-workflow-ci.py
@@ -503,7 +503,7 @@ def build_and_test_macos(id: str, name: str, needs: List[str], attempts: int = 3
                 f'          python --version\n'
                 f'\n'
                 f'          python -m pip install -U pip\n'
-                f'          if [[ ${{TENSORFLOW}} == tensorflow*==1.* ]] || [[ ${{TENSORFLOW_PACKAGE}} == tensorflow*==2.[012345].* ]]; then h5py="h5py<3"; fi\n'
+                f'          if [[ ${{TENSORFLOW}} == 1.* ]] || [[ ${{TENSORFLOW}} == 2.[012345].* ]]; then h5py="h5py<3"; fi\n'
                 f'          pip install tensorflow==${{TENSORFLOW}} keras==${{KERAS}} ${{h5py:-}}\n'
                 f'          pip install torch==${{PYTORCH}} pytorch_lightning==${{PYTORCH_LIGHTNING}} torchvision==${{TORCHVISION}}\n'
                 f'          pip install mxnet==${{MXNET}}\n'

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -3414,7 +3414,7 @@ jobs:
           python --version
 
           python -m pip install -U pip
-          if [[ ${TENSORFLOW} == tensorflow*==1.* ]] || [[ ${TENSORFLOW_PACKAGE} == tensorflow*==2.[012345].* ]]; then h5py="h5py<3"; fi
+          if [[ ${TENSORFLOW} == 1.* ]] || [[ ${TENSORFLOW} == 2.[012345].* ]]; then h5py="h5py<3"; fi
           pip install tensorflow==${TENSORFLOW} keras==${KERAS} ${h5py:-}
           pip install torch==${PYTORCH} pytorch_lightning==${PYTORCH_LIGHTNING} torchvision==${TORCHVISION}
           pip install mxnet==${MXNET}