Skip to content

Commit 1343d5a

Browse files
authored
feat: Add hostname information when requesting model model api requests (#9649)
1 parent 1ddeecd commit 1343d5a

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

label_studio/ml/api_connector.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -201,6 +201,7 @@ def train(self, project, use_ground_truth=False):
201201
'annotations': tasks_ser,
202202
'project': self._create_project_uid(project),
203203
'label_config': project.label_config,
204+
'hostname': settings.HOSTNAME if settings.HOSTNAME else ('http://localhost:' + settings.INTERNAL_PORT),
204205
'params': {'login': project.task_data_login, 'password': project.task_data_password},
205206
}
206207
return self._request('train', request, verbose=False, timeout=TIMEOUT_PREDICT)
@@ -213,6 +214,7 @@ def _prep_prediction_req(self, tasks, project, context=None):
213214
'params': {
214215
'login': project.task_data_login,
215216
'password': project.task_data_password,
217+
'hostname': settings.HOSTNAME if settings.HOSTNAME else ('http://localhost:' + settings.INTERNAL_PORT),
216218
'context': context,
217219
},
218220
}

0 commit comments

Comments
 (0)