Skip to content

Commit

Permalink
return label raster info (#74)
Browse files Browse the repository at this point in the history
* updated

* format
marshHawk4 authored Oct 24, 2024
1 parent 596e420 commit 1aa36a1
Showing 4 changed files with 104 additions and 103 deletions.
5 changes: 3 additions & 2 deletions cdr_schemas/cdr_responses/prospectivity.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
from datetime import datetime
from typing import List, Union
from typing import List, Optional, Union

from geojson_pydantic import LineString, Point, Polygon
from pydantic import BaseModel, ConfigDict, Field
@@ -57,6 +57,7 @@ class CreateVectorProcessDataLayer(BaseModel):
class ProcessedDataLayer(BaseModel):
layer_id: str = Field(description="Layer id")
download_url: str = Field(description="Download url")
label_raster: Optional[bool] = Field(description="Label raster")


class ProspectModelMetaData(BaseModel):
@@ -70,7 +71,7 @@ class ProspectModelMetaData(BaseModel):
model_type: str
train_config: Union[SOMTrainConfig, NeuralNetUserOptions]
evidence_layers: List[ProcessedDataLayer] = Field(
description="Processed data layer ids."
description="Processed data layer info."
)

model_config = ConfigDict(protected_namespaces=())
3 changes: 1 addition & 2 deletions cdr_schemas/prospectivity_input.py
Original file line number Diff line number Diff line change
@@ -150,8 +150,7 @@ class SaveProcessedDataLayer(BaseModel):
transform_methods: TranformMethods = Field(
default_factory=list, description="Transformation methods used"
)
event_id:str = Field(default= "", description="ID of the cma")

event_id: str = Field(default="", description="ID of the cma")


class DefineVectorProcessDataLayer(BaseModel):
197 changes: 99 additions & 98 deletions poetry.lock

Large diffs are not rendered by default.

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.11"
version = "0.4.13"
description = "CDR Schemas"
authors = []
readme = "README.md"

0 comments on commit 1aa36a1

Please sign in to comment.