Skip to content

Commit 2e3137b

Browse files
authored
[PBMP-7171] Use async_reporting by default to speed up response (#1398)
require new tracking-agent https://github.com/datarobot/tracking-agent/pull/3228
1 parent 7d8238c commit 2e3137b

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

custom_model_runner/datarobot_drum/drum/language_predictors/base_language_predictor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -185,6 +185,7 @@ def _configure_mlops(self):
185185
mlops_service_url=self._params["external_webserver_url"],
186186
mlops_api_token=self._params["api_token"],
187187
)
188+
self._mlops.set_async_reporting()
188189

189190
def get_prompt_column_name(self):
190191
if not self._params.get("deployment_id", None):

0 commit comments

Comments
 (0)