Skip to content

Commit b26b15f

Browse files
authored
Update model_evaluator.py
1 parent 0a6277a commit b26b15f

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tools/accuracy_checker/accuracy_checker/evaluators/model_evaluator.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,6 @@
3535
from .base_evaluator import BaseEvaluator
3636
from .quantization_model_evaluator import create_dataset_attributes
3737
from ..metrics.metric_profiler import write_summary_result
38-
from ..representation.base_representation import detach_representation
3938

4039
# pylint: disable=W0223
4140
class ModelEvaluator(BaseEvaluator):
@@ -669,9 +668,9 @@ def _configure_metrics(self, config, output_callback):
669668
ignore_metric_reference)
670669

671670
@staticmethod
672-
def store_predictions(stored_predictions, predictions, mode='ab'):
671+
def store_predictions(stored_predictions, predictions):
673672
# since at the first time file does not exist and then created we can not use it as a pathlib.Path object
674-
with open(stored_predictions, mode) as content:
673+
with open(stored_predictions, 'ab') as content:
675674
pickle.dump(predictions, content)
676675

677676
@staticmethod

0 commit comments

Comments
 (0)