From 9e5a45605c45f1dc59790609ec73f4a6ca986a7b Mon Sep 17 00:00:00 2001
From: Arun Vasudevan <avasudevan@homeaway.com>
Date: Tue, 25 Aug 2020 15:06:00 -0500
Subject: [PATCH] Update to Dataset

---
 .../src/main/pegasus/com/linkedin/ml/metadata/BaseData.pdl  | 6 +++---
 .../pegasus/com/linkedin/ml/metadata/EvaluationData.pdl     | 2 +-
 .../main/pegasus/com/linkedin/ml/metadata/TrainingData.pdl  | 2 +-
 3 files changed, 5 insertions(+), 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
index db7c6c8290e70..a84c0372e92fc 100644
--- 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
index 2aa614923da54..0f029b61d4a93 100644
--- 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
index 3b77c91d21a76..d647192c711dc 100644
--- 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]
 }