Upgrade pydantic

doccano · Jul 20, 2023 · 2759143 · 2759143
1 parent 6352807
commit 2759143
Show file tree

Hide file tree

Showing 2 changed files with 127 additions and 31 deletions.
diff --git a/Pipfile.lock b/Pipfile.lock
diff --git a/auto_labeling_pipeline/models.py b/auto_labeling_pipeline/models.py
@@ -33,6 +33,8 @@ def get_all_subclasses(cls):
             return all_subclasses
 
         for subclass in get_all_subclasses(RequestModel):
+            if not hasattr(subclass, "Config"):
+                continue
             if subclass.Config.title == model_name:
                 return subclass
         raise NameError(f'{model_name} is not found.')
@@ -84,7 +86,7 @@ class GCPEntitiesRequestModel(RequestModel):
 
     class Config:
         title = 'GCP Entity Analysis'
-        schema_extra = {
+        json_schema_extra = {
             # https://cloud.google.com/natural-language/docs/reference/rest/v1/Entity#Type
             'types': [
                 'UNKNOWN', 'PERSON', 'LOCATION', 'ORGANIZATION', 'EVENT', 'WORK_OF_ART',
@@ -152,7 +154,7 @@ class AmazonComprehendSentimentRequestModel(AmazonComprehendRequestModel):
 
     class Config:
         title = 'Amazon Comprehend Sentiment Analysis'
-        schema_extra = {
+        json_schema_extra = {
             # https://docs.aws.amazon.com/comprehend/latest/dg/how-sentiment.html
             'types': [
                 'POSITIVE', 'NEGATIVE', 'NEUTRAL', 'MIXED'
@@ -175,7 +177,7 @@ class AmazonComprehendEntityRequestModel(AmazonComprehendRequestModel):
 
     class Config:
         title = 'Amazon Comprehend Entity Recognition'
-        schema_extra = {
+        json_schema_extra = {
             # https://docs.aws.amazon.com/comprehend/latest/dg/how-entities.html
             'types': [
                 'PERSON', 'LOCATION', 'ORGANIZATION', 'COMMERCIAL_ITEM',
@@ -200,7 +202,7 @@ class AmazonComprehendPIIEntityRequestModel(AmazonComprehendRequestModel):
 
     class Config:
         title = 'Amazon Comprehend PII Entity Recognition'
-        schema_extra = {
+        json_schema_extra = {
             # https://docs.aws.amazon.com/comprehend/latest/dg/how-pii.html
             'types': [
                 'BANK_ACCOUNT_NUMBER', 'BANK_ROUTING', 'CREDIT_DEBIT_NUMBER',