changed to slow test type

JohnSnowLabs · Mar 11, 2024 · 5dc52d8 · 5dc52d8
1 parent 1ccf943
commit 5dc52d8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/test/scala/com/johnsnowlabs/nlp/annotators/seq2seq/CPMTestSpec.scala b/src/test/scala/com/johnsnowlabs/nlp/annotators/seq2seq/CPMTestSpec.scala
@@ -18,14 +18,14 @@ package com.johnsnowlabs.nlp.annotators.seq2seq
 
 import com.johnsnowlabs.nlp.base.DocumentAssembler
 import com.johnsnowlabs.nlp.util.io.ResourceHelper
-import com.johnsnowlabs.tags.FastTest
+import com.johnsnowlabs.tags.{FastTest, SlowTest}
 import org.apache.spark.ml.Pipeline
 import org.scalatest.flatspec.AnyFlatSpec
 import org.sparkproject.dmg.pmml.True
 
 class CPMTestSpec extends AnyFlatSpec {
 
-  "minicpm" should "should handle temperature=0 correctly and not crash when predicting more than 1 element with doSample=True" taggedAs FastTest in {
+  "minicpm" should "should handle temperature=0 correctly and not crash when predicting more than 1 element with doSample=True" taggedAs SlowTest in {
     // Even tough the Paper states temperature in interval [0,1), using temperature=0 will result in division by 0 error.
     // Also DoSample=True may result in infinities being generated and distFiltered.length==0 which results in exception if we don't return 0 instead internally.
     val testData = ResourceHelper.spark