From b16925378340d72d44129fff5064e1abcbf1995d Mon Sep 17 00:00:00 2001 From: Anton Gerdelan Date: Mon, 19 Feb 2024 10:33:53 +0000 Subject: [PATCH] fixed formatting pre-commit check --- src/brevitas/export/onnx/standard/qoperator/manager.py | 1 + src/brevitas/export/torch/qoperator/manager.py | 1 + 2 files changed, 2 insertions(+) diff --git a/src/brevitas/export/onnx/standard/qoperator/manager.py b/src/brevitas/export/onnx/standard/qoperator/manager.py index f45bebc9e..3ec1202a4 100644 --- a/src/brevitas/export/onnx/standard/qoperator/manager.py +++ b/src/brevitas/export/onnx/standard/qoperator/manager.py @@ -28,6 +28,7 @@ from .handler.parameter import StdQOpONNXQuantConv2dHandler from .handler.parameter import StdQOpONNXQuantLinearHandler + class StdQOpONNXManager(StdONNXBaseManager): target_name = 'StdQOpONNX' diff --git a/src/brevitas/export/torch/qoperator/manager.py b/src/brevitas/export/torch/qoperator/manager.py index 0215b7a59..f244bbcf7 100644 --- a/src/brevitas/export/torch/qoperator/manager.py +++ b/src/brevitas/export/torch/qoperator/manager.py @@ -21,6 +21,7 @@ from .handler.parameter import PytorchQuantConv2dHandler from .handler.parameter import PytorchQuantLinearHandler + class TorchQOpManager(BaseManager): target_name = 'torch'