added

DARPA-CRITICALMAAS · Oct 30, 2024 · 0674c43 · 0674c43
1 parent c7cf76d
commit 0674c43
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cdr_schemas/cdr_responses/prospectivity.py b/cdr_schemas/cdr_responses/prospectivity.py
@@ -73,7 +73,7 @@ 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]
+    train_config: Union[SOMTrainConfig, NeuralNetUserOptions, RFUserOptions]
     evidence_layers: List[ProcessedDataLayer] = Field(
         description="Processed data layer info."
     )