diff --git a/src/deepsparse/pipeline_config.py b/src/deepsparse/pipeline_config.py index 57bf07fa97..f76ba071d4 100644 --- a/src/deepsparse/pipeline_config.py +++ b/src/deepsparse/pipeline_config.py @@ -14,7 +14,7 @@ from typing import Any, Dict, List, Optional -from pydantic import BaseModel, Field, ConfigDict +from pydantic import BaseModel, ConfigDict, Field from deepsparse.operators.engine_operator import DEEPSPARSE_ENGINE @@ -86,6 +86,4 @@ class PipelineConfig(BaseModel): ) # override name spaces due to model_ warnings in pydantic 2.X - model_config = ConfigDict( - protected_namespaces=() - ) + model_config = ConfigDict(protected_namespaces=()) diff --git a/src/deepsparse/server/server.py b/src/deepsparse/server/server.py index 1eff41f917..ca9b17caed 100644 --- a/src/deepsparse/server/server.py +++ b/src/deepsparse/server/server.py @@ -53,9 +53,7 @@ class ModelMetaData(BaseModel): model_path: str # override name spaces due to model_ warnings in pydantic 2.X - model_config = ConfigDict( - protected_namespaces=() - ) + model_config = ConfigDict(protected_namespaces=()) # For deepsparse endpoints, we bind the `predict`` and `predict_from_files` functions to