Add back UNIQUENESS_ID

awslabs · Aug 31, 2024 · 54d4d78 · 54d4d78
1 parent 542dcd6
commit 54d4d78
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/scala/com/amazon/deequ/VerificationResult.scala b/src/main/scala/com/amazon/deequ/VerificationResult.scala
@@ -97,7 +97,8 @@ object VerificationResult {
 
     val columnNamesToMetrics: Map[String, Column] = verificationResultToColumn(verificationResult)
 
-    data.withColumns(columnNamesToMetrics)
+    val dataWithID = data.withColumn(UNIQUENESS_ID, monotonically_increasing_id())
+    dataWithID.withColumns(columnNamesToMetrics).drop(UNIQUENESS_ID)
   }
 
   def checkResultsAsJson(verificationResult: VerificationResult,