From 63bf0da39db675131388f3eefb25704c9083bacc Mon Sep 17 00:00:00 2001
From: Maziyar Panahi <maziyar.panahi@iscpif.fr>
Date: Tue, 12 Mar 2024 10:14:55 +0100
Subject: [PATCH] Fix bad access column

---
 src/main/scala/Main.scala | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/main/scala/Main.scala b/src/main/scala/Main.scala
index e9b7793..16d51cc 100644
--- 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)
 
   }