diff --git a/prepare/src/main/guessNNprepare/mains/ComputeConcepts.scala b/prepare/src/main/guessNNprepare/mains/ComputeConcepts.scala index 42215a9..5034e1d 100644 --- a/prepare/src/main/guessNNprepare/mains/ComputeConcepts.scala +++ b/prepare/src/main/guessNNprepare/mains/ComputeConcepts.scala @@ -9,6 +9,7 @@ import org.apache.jena.sparql.core.Var import org.apache.jena.sparql.engine.binding.BindingFactory import org.json.{JSONArray, JSONObject, JSONTokener} +import java.io.PrintWriter import java.nio.file.{Files, Path, Paths} import java.util.concurrent.atomic.AtomicBoolean import scala.jdk.CollectionConverters.SeqHasAsJava @@ -97,10 +98,10 @@ object ComputeConcepts extends MainCommand { private def processConcept(concept: JSONObject): JSONObject = { val result = new JSONObject() result.put("extensionalDistance", concept.getInt("extensionalDistance")) - val objects: Seq[JSONObject] = Utils.JSONArrayElementsAs(concept.getJSONArray("answers")) + val objects: Seq[String] = Utils.JSONArrayElementsAs[JSONArray](concept.getJSONArray("answers")) + .map(answerRow => answerRow.getString(0)) result.put("properSize", objects.size) result.put("elements", new JSONArray(objects.asJava)) - result } }