From 736106539f9469f5c59f75fa3b39b72244e0a5b9 Mon Sep 17 00:00:00 2001 From: tmartins Date: Thu, 2 Jun 2022 06:36:33 -0300 Subject: [PATCH] remove ml module dependency from application module --- vespa/application.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/vespa/application.py b/vespa/application.py index 5ec5fa5f..04378ca5 100644 --- a/vespa/application.py +++ b/vespa/application.py @@ -19,7 +19,6 @@ from time import sleep from vespa.io import VespaQueryResponse, VespaResponse -from vespa.ml import TextTask from vespa.query import QueryModel from vespa.evaluation import EvalMetric from vespa.package import ApplicationPackage @@ -1166,7 +1165,7 @@ def predict(self, x, model_id, function_name="output_0"): model = self.get_model_from_application_package(model_id) encoded_tokens = model.create_url_encoded_tokens(x=x) with VespaSync(self) as sync_app: - return TextTask.parse_vespa_prediction( + return model.parse_vespa_prediction( sync_app.predict( model_id=model_id, function_name=function_name,