chore: fix flaky serialization fuzzing test

microsoft · Aug 23, 2021 · fdf7562 · fdf7562
1 parent f5b9c5e
commit fdf7562
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cognitive/src/test/scala/com/microsoft/ml/spark/cognitive/split1/ComputerVisionSuite.scala b/cognitive/src/test/scala/com/microsoft/ml/spark/cognitive/split1/ComputerVisionSuite.scala
@@ -442,7 +442,7 @@ class TagImageSuite extends TransformerFuzzing[TagImage] with CognitiveKey with
   }
 
   override def assertDFEq(df1: DataFrame, df2: DataFrame)(implicit eq: Equality[DataFrame]): Unit = {
-    super.assertDFEq(df1.select("tags.tags"), df2.select("tags.tags"))(eq)
+    super.assertDFEq(df1.select("tags.tags.name"), df2.select("tags.tags.name"))(eq)
   }
 
   override def testObjects(): Seq[TestObject[TagImage]] =