diff --git a/cdr_schemas/cdr_responses/prospectivity.py b/cdr_schemas/cdr_responses/prospectivity.py index 0d598e6..bccb40c 100644 --- a/cdr_schemas/cdr_responses/prospectivity.py +++ b/cdr_schemas/cdr_responses/prospectivity.py @@ -1,7 +1,7 @@ from datetime import datetime from typing import List, Union -from pydantic import BaseModel, Field +from pydantic import BaseModel, ConfigDict, Field from cdr_schemas.prospectivity_input import ( CreateCriticalMineralAssessment, @@ -41,3 +41,5 @@ class ProspectModelMetaData(BaseModel): model_type: str train_config: Union[SOMTrainConfig, NeuralNetUserOptions] evidence_layers: List[CreateProcessDataLayer] + + model_config = ConfigDict(protected_namespaces=()) diff --git a/cdr_schemas/prospectivity_input.py b/cdr_schemas/prospectivity_input.py index e61dad2..ffe161f 100644 --- a/cdr_schemas/prospectivity_input.py +++ b/cdr_schemas/prospectivity_input.py @@ -3,7 +3,7 @@ from typing import List, Optional, Union from geojson_pydantic import MultiPolygon -from pydantic import BaseModel, Field +from pydantic import BaseModel, ConfigDict, Field from cdr_schemas.prospectivity_models import ( NeuralNetUserOptions, @@ -82,6 +82,8 @@ class ProspectivityOutputLayer(BaseModel): cma_id: str = Field(description="id of the cma") title: str = Field(description="Title for prospectivity layer") + model_config = ConfigDict(protected_namespaces=()) + # MTRI to CDR: # send to cdr to create new cma. Will be associated with template raster uploaded @@ -123,6 +125,7 @@ class SaveProcessedDataLayer(BaseModel): transform_methods: TranformMethods = Field( default="", description="Transformation method used" ) + model_config = ConfigDict(protected_namespaces=()) # MTRI UI to CDR: @@ -139,6 +142,7 @@ class CreateProspectModelMetaData(BaseModel): evidence_layers: List[DefineProcessDataLayer] = Field( description="Datasource and preprocess steps" ) + model_config = ConfigDict(protected_namespaces=()) class DataSource(BaseModel): diff --git a/pyproject.toml b/pyproject.toml index ffceb47..56eaa20 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "cdr_schemas" -version = "0.4.5" +version = "0.4.8" description = "CDR Schemas" authors = [] readme = "README.md"