Skip to content

Commit 86ea8f6

Browse files
authored
Add unit tests for nncf quantization functions (#2526)
### Details - Added unit tests for quantization and accuracy control functions - Files tested: * [nncf/quantization/algorithms/accuracy_control/evaluator.py](https://app.codecov.io/gh/openvinotoolkit/nncf/blob/develop/nncf/quantization/algorithms/accuracy_control/evaluator.py) * [nncf/quantization/algorithms/accuracy_control/subset_selection.py](https://app.codecov.io/gh/openvinotoolkit/nncf/blob/develop/nncf/quantization/algorithms/accuracy_control/subset_selection.py) * [nncf/quantization/algorithms/accuracy_control/openvino_backend.py](https://app.codecov.io/gh/openvinotoolkit/nncf/blob/develop/nncf/quantization/algorithms/accuracy_control/openvino_backend.py) * [nncf/quantization/algorithms/accuracy_control/ranker.py](https://app.codecov.io/gh/openvinotoolkit/nncf/blob/develop/nncf/quantization/algorithms/accuracy_control/ranker.py) * [nncf/quantization/algorithms/accuracy_control/algorithm.py](https://app.codecov.io/gh/openvinotoolkit/nncf/blob/develop/nncf/quantization/algorithms/accuracy_control/algorithm.py) - For [ranker.py](https://github.com/openvinotoolkit/nncf/blob/develop/nncf/quantization/algorithms/accuracy_control/ranker.py), I am not sure how to test the case with multithreading, so I have left that out for now. - For [algorithm.py](https://github.com/openvinotoolkit/nncf/blob/develop/nncf/quantization/algorithms/accuracy_control/algorithm.py), I am not sure how to test some of the more complex functions like `apply`, so I have left that out for now. - Addresses #2496
1 parent 38be944 commit 86ea8f6

File tree

4 files changed

+826
-17
lines changed

4 files changed

+826
-17
lines changed

nncf/quantization/algorithms/accuracy_control/evaluator.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ def disable_iteration_count(self) -> None:
101101
"""
102102
self._enable_iteration_count = False
103103

104-
def is_metric_mode(self) -> bool:
104+
def is_metric_mode(self) -> Optional[bool]:
105105
"""
106106
Returns mode of `Evaluator`.
107107

0 commit comments

Comments
 (0)