Update to Dataset

arunvasudevan · Sep 10, 2020 · 9e5a456 · 9e5a456
1 parent 64c5249
commit 9e5a456
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/metadata-models/src/main/pegasus/com/linkedin/ml/metadata/BaseData.pdl b/metadata-models/src/main/pegasus/com/linkedin/ml/metadata/BaseData.pdl
@@ -8,12 +8,12 @@ import com.linkedin.common.DatasetUrn
 record BaseData {
 
   /**
-   * What datasets were used in the MLModel?
+   * What dataset were used in the MLModel?
    */
-  datasets: optional DatasetUrn
+  dataset: DatasetUrn
 
   /**
-   * Why were these datasets chosen?
+   * Why was this dataset chosen?
    */
   motivation: optional string
 

diff --git a/metadata-models/src/main/pegasus/com/linkedin/ml/metadata/EvaluationData.pdl b/metadata-models/src/main/pegasus/com/linkedin/ml/metadata/EvaluationData.pdl
@@ -10,5 +10,5 @@ record EvaluationData {
     /**
      * Details on the dataset(s) used for the quantitative analyses in the MLModel
      */
-    evaluationData: optional array[BaseData]
+    evaluationData: array[BaseData]
 }
diff --git a/metadata-models/src/main/pegasus/com/linkedin/ml/metadata/TrainingData.pdl b/metadata-models/src/main/pegasus/com/linkedin/ml/metadata/TrainingData.pdl
@@ -10,5 +10,5 @@ record TrainingData {
     /**
     * Details on the dataset(s) used for training the MLModel
     */
-    trainingData: optional array[BaseData]
+    trainingData: array[BaseData]
 }