From 073965c47e7e31f1255fd461cd34ee19917d78bb Mon Sep 17 00:00:00 2001 From: mertalev <101130780+mertalev@users.noreply.github.com> Date: Fri, 2 Feb 2024 23:35:05 -0500 Subject: [PATCH] formatting --- machine-learning/app/models/base.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/machine-learning/app/models/base.py b/machine-learning/app/models/base.py index 7f707abfee..836c4a5012 100644 --- a/machine-learning/app/models/base.py +++ b/machine-learning/app/models/base.py @@ -12,6 +12,7 @@ from huggingface_hub import snapshot_download from onnx.shape_inference import infer_shapes_path from onnx.tools.update_model_dims import update_inputs_outputs_dims from typing_extensions import Buffer + import ann.ann from app.models.constants import STATIC_INPUT_PROVIDERS, SUPPORTED_PROVIDERS @@ -145,7 +146,7 @@ class InferenceModel(ABC): # check_model gets called in update_inputs_outputs_dims check_model = onnx.checker.check_model try: - + def check_model_stub(*args: Any, **kwargs: Any) -> None: pass