Merge pull request #293 from vipyrsec/revert-revert-br/gh166

Update deprecated Pydantic config
vipyrsec · Jul 20, 2024 · 80b0752 · 80b0752
2 parents ad896db + 8b0547e
commit 80b0752
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/mainframe/models/schemas.py b/src/mainframe/models/schemas.py
@@ -2,7 +2,7 @@
 from enum import Enum
 from typing import Any, Optional
 
-from pydantic import BaseModel, Field, field_serializer
+from pydantic import BaseModel, Field, field_serializer, ConfigDict
 
 from .orm import Scan
 
@@ -85,12 +85,11 @@ class PackageSpecifier(BaseModel):
     version: A str of the package version to scan.
     """
 
+    model_config = ConfigDict(frozen=True)
+
     name: str
     version: str
 
-    class Config:
-        frozen = True
-
 
 class ReportPackageBody(PackageSpecifier):
     recipient: Optional[str]