Skip to content

Commit

Permalink
warnings (#65)
Browse files Browse the repository at this point in the history
* warnings

* warnings

* warnings
  • Loading branch information
marshHawk4 authored Sep 12, 2024
1 parent 0650372 commit b9f0701
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 3 deletions.
4 changes: 3 additions & 1 deletion cdr_schemas/cdr_responses/prospectivity.py
Original file line number Diff line number Diff line change
@@ -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,
Expand Down Expand Up @@ -41,3 +41,5 @@ class ProspectModelMetaData(BaseModel):
model_type: str
train_config: Union[SOMTrainConfig, NeuralNetUserOptions]
evidence_layers: List[CreateProcessDataLayer]

model_config = ConfigDict(protected_namespaces=())
6 changes: 5 additions & 1 deletion cdr_schemas/prospectivity_input.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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:
Expand All @@ -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):
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[tool.poetry]
name = "cdr_schemas"
version = "0.4.5"
version = "0.4.8"
description = "CDR Schemas"
authors = []
readme = "README.md"
Expand Down

0 comments on commit b9f0701

Please sign in to comment.