Skip to content

Commit 6e6350f

Browse files
authored
Move exporter E2E tests to pytest (#1476)
* Move exporter E2E tests to pytest * Move tests to MCTQ-nightly * Disable outputs compare check for now, until further debug
1 parent 14516ce commit 6e6350f

17 files changed

Lines changed: 391 additions & 817 deletions

.github/workflows/run_keras_tests.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,8 @@ jobs:
2020
with:
2121
python-version: ${{ inputs.python-version }}
2222
- name: Install dependencies
23-
run: |
23+
run: |
24+
sed -i "/mct-quantizers/c\mct-quantizers-nightly" requirements.txt
2425
python -m pip install --upgrade pip
2526
pip install -r requirements.txt
2627
pip install tensorflow==${{ inputs.tf-version }} sony-custom-layers

.github/workflows/run_pytorch_tests.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ jobs:
2727
mv temp_requirements.txt requirements.txt
2828
- name: Install dependencies
2929
run: |
30+
sed -i "/mct-quantizers/c\mct-quantizers-nightly" requirements.txt
3031
python -m pip install --upgrade pip
3132
pip install -r requirements.txt
3233
pip install torch==${{ inputs.torch-version }} torchvision onnx onnxruntime "onnxruntime-extensions<0.14"

model_compression_toolkit/exporter/model_exporter/pytorch/fakely_quant_onnx_pytorch_exporter.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,6 @@
1919

2020
from mct_quantizers import PytorchActivationQuantizationHolder, PytorchQuantizationWrapper
2121

22-
from model_compression_toolkit.core.pytorch.reader.node_holders import DummyPlaceHolder
2322
from model_compression_toolkit.verify_packages import FOUND_ONNX
2423
from model_compression_toolkit.logger import Logger
2524
from model_compression_toolkit.core.pytorch.utils import to_torch_tensor

tests/pytorch_tests/exporter_tests/__init__.py

Lines changed: 0 additions & 13 deletions
This file was deleted.

tests/pytorch_tests/exporter_tests/base_pytorch_export_test.py

Lines changed: 0 additions & 88 deletions
This file was deleted.

tests/pytorch_tests/exporter_tests/base_pytorch_onnx_export_test.py

Lines changed: 0 additions & 104 deletions
This file was deleted.

tests/pytorch_tests/exporter_tests/custom_ops_tests/__init__.py

Lines changed: 0 additions & 14 deletions
This file was deleted.

tests/pytorch_tests/exporter_tests/custom_ops_tests/test_export_lut_symmetric_onnx_quantizers.py

Lines changed: 0 additions & 76 deletions
This file was deleted.

0 commit comments

Comments
 (0)