Use mkFast on Grounder

ml4ai · Apr 29, 2024 · 3b27cc0 · 3b27cc0
1 parent 6bb84e2
commit 3b27cc0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...scala/src/main/scala/org/ml4ai/skema/text_reading/grounding/scala_grounders/Adapter.scala b/...scala/src/main/scala/org/ml4ai/skema/text_reading/grounding/scala_grounders/Adapter.scala
@@ -31,9 +31,9 @@ import scala.io.Source
   *                             a concept that is outside this)
   */
 class ScalaGroundersAdapter(groundingConcepts: Seq[GroundingConcept]) extends Grounder {
-  lazy val grounder = SequentialGrounder()
+  lazy val concepts = groundingConcepts.map(fromConceptToDKG)
+  lazy val grounder = SequentialGrounder().mkFast(concepts)
   def groundingCandidates(texts: Seq[String], k: Int): Seq[Seq[GroundingCandidate]] = {
-    val concepts = groundingConcepts.map(fromConceptToDKG)
     texts.map { text =>
       // TODO Maybe provide additional context (useful for NeuralGrounder)
       grounder.ground(text, None, concepts, k)