add pytest-mock to CI dependencies

sony · Jan 26, 2025 · 1a3e384 · 1a3e384
1 parent 6dc2e21
commit 1a3e384
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/.github/workflows/run_keras_tests.yml b/.github/workflows/run_keras_tests.yml
@@ -23,8 +23,9 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           pip install -r requirements.txt
-          pip install tensorflow==${{ inputs.tf-version }} sony-custom-layers pytest
-
+          pip install tensorflow==${{ inputs.tf-version }} sony-custom-layers 
+          pip install pytest pytest-mock
+          pip check
       - name: Run unittests
         run: |
           python -m unittest discover tests/keras_tests -v

diff --git a/.github/workflows/run_pytorch_tests.yml b/.github/workflows/run_pytorch_tests.yml
@@ -24,7 +24,8 @@ jobs:
           python -m pip install --upgrade pip
           pip install -r requirements.txt          
           pip install torch==${{ inputs.torch-version }} torchvision onnx onnxruntime onnxruntime-extensions
-          pip install pytest
+          pip install pytest pytest-mock
+          pip check
       - name: Run unittests
         run: |
           python -m unittest discover tests/pytorch_tests -v