Merge pull request #106 from microsoft/Tarun/Pydantic

Migrate Pydantic V2

Author: arunsu

azureml_inference_server_http/server/config.py

@@ -145,7 +145,7 @@ class AMLInferenceServerConfig(BaseSettings):
     debug_port: Optional[int] = pydantic.Field(default=None, alias="AZUREML_DEBUG_PORT")
 
     # Check if extra keys are there in the config file
-    @pydantic.root_validator(pre=True)
+    @pydantic.model_validator(mode="before")
     def check_extra_keys(cls, values: Dict[str, Any]):
         supported_keys = alias_mapping.values()
         extra_keys = []

tests/server/conftest.py

@@ -64,11 +64,11 @@ def app_appinsights(config):
 
 @pytest.fixture()
 def config():
-    backup_config = server_config.copy()
+    backup_config = server_config.model_copy()
     try:
         yield server_config
     finally:
-        for field in server_config.__fields__:
+        for field in server_config.model_fields:
             setattr(server_config, field, getattr(backup_config, field))