Fix bad access column

maziyarpanahi · Mar 12, 2024 · 63bf0da · 63bf0da
1 parent d0ba604
commit 63bf0da
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/main/scala/Main.scala b/src/main/scala/Main.scala
@@ -76,8 +76,7 @@ object Main {
     val mpnet_lightpipeline = new LightPipeline(pipeline.fit(testData))
     mpnet_lightpipeline.annotate(
       "Google has announced the release of a beta version of the popular TensorFlow machine learning library")
-    mpnet_lightpipeline.transform(testData).count()
-    mpnet_lightpipeline.transform(testData).select("entities").show(false)
+    mpnet_lightpipeline.transform(testData).select("mpnet").count().show(false)
 
   }