Add extra schema for TA3 models (#54)

* updated * these are saved * save
DARPA-CRITICALMAAS · Jul 30, 2024 · 2ac96f5 · 2ac96f5
1 parent ce755f8
commit 2ac96f5
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/cdr_schemas/cdr_responses/prospectivity.py b/cdr_schemas/cdr_responses/prospectivity.py
@@ -21,9 +21,6 @@ class DataSource(CreateDataSource):
 
 
 class CreateProcessDataLayer(BaseModel):
- cma: CriticalMineralAssessment = Field(
- description="CMA with all information needed for processing"
- )
  data_source: DataSource = Field(description="Data source to create this layer")
  title: str = Field(description="Title to use for processed layer")
  transform_methods: TranformMethods = Field(
@@ -39,5 +36,6 @@ class ProspectModelMetaData(BaseModel):
 
  model_run_id: str = Field(description="CDR id of the model run")
  cma: CriticalMineralAssessment = Field(description="CMA info")
+ model_type: str
  train_config: Union[SOMTrainConfig, NeuralNetUserOptions]
  evidence_layers: List[CreateProcessDataLayer]
diff --git a/cdr_schemas/prospectivity_input.py b/cdr_schemas/prospectivity_input.py
@@ -131,6 +131,7 @@ class CreateProspectModelMetaData(BaseModel):
  author: str = ""
  date: str = ""
  organization: str = ""
+ model_type: str
  train_config: Union[SOMTrainConfig, NeuralNetUserOptions]
  evidence_layers: List[DefineProcessDataLayer] = Field(
  description="Datasource and preprocess steps"

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cdr_schemas"
-version = "0.4.0"
+version = "0.4.1"
 description = "CDR Schemas"
 authors = []
 readme = "README.md"