Skip to content

Commit b4000d3

Browse files
committed
Fix compilation error
1 parent 3054d4c commit b4000d3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/MPNetForTokenClassification.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -373,7 +373,7 @@ trait ReadMPNetForTokenDLModel extends ReadOnnxModel {
373373
case TensorFlow.name =>
374374
throw new NotImplementedError("Tensorflow models are not supported.")
375375
case ONNX.name =>
376-
val onnxWrapper = OnnxWrapper.read(localModelPath, zipped = false, useBundle = true)
376+
val onnxWrapper = OnnxWrapper.read(spark, localModelPath, zipped = false, useBundle = true)
377377
annotatorModel
378378
.setModelIfNotSet(spark, Some(onnxWrapper))
379379
case _ =>

0 commit comments

Comments
 (0)