Merge pull request #43 from clulab/kwalcock/maxtokens

Fix test
clulab · Sep 17, 2023 · 69a82e8 · 69a82e8
2 parents 8d5d7c4 + fd8e3ae
commit 69a82e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.../src/test/scala/org/clulab/scala_transformers/encoder/timer/TokenClassifierTimerApp.scala b/.../src/test/scala/org/clulab/scala_transformers/encoder/timer/TokenClassifierTimerApp.scala
@@ -9,7 +9,7 @@ object TokenClassifierTimerApp extends App {
   val fileName = args.lift(0).getOrElse("../sentences.txt")
 
   class TimedTokenClassifier(tokenClassifier: TokenClassifier) extends TokenClassifier(
-    tokenClassifier.encoder, tokenClassifier.tasks, tokenClassifier.tokenizer
+    tokenClassifier.encoder, tokenClassifier.maxTokens, tokenClassifier.tasks, tokenClassifier.tokenizer
   ) {
     val tokenizeTimer = Timers.getOrNew("Tokenizer")
     val forwardTimer = Timers.getOrNew("Encoder.forward")