diff --git a/cognitive/src/main/scala/com/microsoft/ml/spark/cognitive/FormRecognizer.scala b/cognitive/src/main/scala/com/microsoft/ml/spark/cognitive/FormRecognizer.scala index 0484691c66..cfc82209c2 100644 --- a/cognitive/src/main/scala/com/microsoft/ml/spark/cognitive/FormRecognizer.scala +++ b/cognitive/src/main/scala/com/microsoft/ml/spark/cognitive/FormRecognizer.scala @@ -18,7 +18,7 @@ import spray.json._ abstract class FormRecognizerBase(override val uid: String) extends CognitiveServicesBaseNoHandler(uid) with HasCognitiveServiceInput with HasInternalJsonOutputParser with BasicAsyncReply - with HasImageInput with HasSetLocation { + with HasImageInput with HasSetLocation with HasSetLinkedService { override protected def prepareEntity: Row => Option[AbstractHttpEntity] = { r => diff --git a/cognitive/src/main/scala/com/microsoft/ml/spark/cognitive/TextTranslator.scala b/cognitive/src/main/scala/com/microsoft/ml/spark/cognitive/TextTranslator.scala index 2868833f14..b6d2623b6f 100644 --- a/cognitive/src/main/scala/com/microsoft/ml/spark/cognitive/TextTranslator.scala +++ b/cognitive/src/main/scala/com/microsoft/ml/spark/cognitive/TextTranslator.scala @@ -84,7 +84,7 @@ trait TextAsOnlyEntity extends HasTextInput with HasCognitiveServiceInput { abstract class TextTranslatorBase(override val uid: String) extends CognitiveServicesBase(uid) with HasInternalJsonOutputParser with HasCognitiveServiceInput with HasSubscriptionRegion - with HasSetLocation { + with HasSetLocation with HasSetLinkedService { protected val subscriptionRegionHeaderName = "Ocp-Apim-Subscription-Region"