Skip to content

Commit

Permalink
linter
Browse files Browse the repository at this point in the history
  • Loading branch information
AlexanderDokuchaev committed Nov 25, 2022
1 parent e777747 commit 15e5e18
Show file tree
Hide file tree
Showing 5 changed files with 5 additions and 14 deletions.
3 changes: 1 addition & 2 deletions external/anomaly/tasks/openvino.py
Original file line number Diff line number Diff line change
Expand Up @@ -274,8 +274,7 @@ def optimize(
# Training subset does not contain example of anomalous images.
# Anomalous examples from all dataset used to get statistics for quantization.
dataset = DatasetEntity(
items = [item for item in dataset if item.get_shapes_labels()[0].is_anomalous],
purpose = dataset.purpose
items=[item for item in dataset if item.get_shapes_labels()[0].is_anomalous], purpose=dataset.purpose
)

logger.info("Starting POT optimization.")
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -308,9 +308,7 @@ def main():
dataset,
environment.get_model_configuration(),
)
openvino_task.optimize(
OptimizationType.POT, dataset, optimized_model, OptimizationParameters()
)
openvino_task.optimize(OptimizationType.POT, dataset, optimized_model, OptimizationParameters())

logger.info("Get predictions on the validation set")
predicted_validation_dataset = openvino_task.infer(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -252,9 +252,7 @@ def main(args):
dataset,
environment.get_model_configuration(),
)
openvino_task.optimize(
OptimizationType.POT, dataset, optimized_model, OptimizationParameters()
)
openvino_task.optimize(OptimizationType.POT, dataset, optimized_model, OptimizationParameters())

logger.info("Get predictions on the validation set")
predicted_validation_dataset = openvino_task.infer(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -300,9 +300,7 @@ def main(args):
dataset,
environment.get_model_configuration(),
)
openvino_task.optimize(
OptimizationType.POT, dataset, optimized_model, OptimizationParameters()
)
openvino_task.optimize(OptimizationType.POT, dataset, optimized_model, OptimizationParameters())

logger.info("Get predictions on the validation set")
predicted_validation_dataset = openvino_task.infer(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -271,9 +271,7 @@ def main(args):
dataset,
environment.get_model_configuration(),
)
openvino_task.optimize(
OptimizationType.POT, dataset, optimized_model, OptimizationParameters()
)
openvino_task.optimize(OptimizationType.POT, dataset, optimized_model, OptimizationParameters())

logger.info("Get predictions on the validation set")
predicted_validation_dataset = openvino_task.infer(
Expand Down

0 comments on commit 15e5e18

Please sign in to comment.