Skip to content

Commit c2b8d41

Browse files
committed
moved wait work request to background thread
1 parent cabbfa4 commit c2b8d41

File tree

1 file changed

+12
-8
lines changed

1 file changed

+12
-8
lines changed

ads/aqua/modeldeployment/deployment.py

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44

55
import json
66
import shlex
7+
import threading
78
from datetime import datetime, timedelta
89
from typing import Dict, List, Optional
910

@@ -53,7 +54,7 @@
5354
from ads.aqua.modeldeployment.config_loader import (
5455
AquaDeploymentConfig,
5556
ConfigurationItem,
56-
ModelDeploymentConfigSummary
57+
ModelDeploymentConfigSummary,
5758
)
5859
from ads.aqua.modeldeployment.constants import DEFAULT_POLL_INTERVAL, DEFAULT_WAIT_TIME
5960
from ads.aqua.modeldeployment.entities import (
@@ -64,7 +65,6 @@
6465
)
6566
from ads.aqua.modeldeployment.model_group_config import ModelGroupConfig
6667
from ads.aqua.modeldeployment.utils import MultiModelDeploymentConfigLoader
67-
from ads.common.decorator.threaded import thread_pool
6868
from ads.common.object_storage_details import ObjectStorageDetails
6969
from ads.common.utils import UNKNOWN, get_log_links
7070
from ads.common.work_request import DataScienceWorkRequest
@@ -730,13 +730,17 @@ def _create_deployment(
730730
f"Aqua model deployment {deployment_id} created for model {aqua_model_id}. Work request Id is {deployment.dsc_model_deployment.workflow_req_id}"
731731
)
732732

733-
thread_pool.submit(
734-
self.get_deployment_status,
735-
deployment_id,
736-
deployment.dsc_model_deployment.workflow_req_id,
737-
model_type,
738-
model_name,
733+
progress_thread = threading.Thread(
734+
target=self.get_deployment_status,
735+
args=(
736+
deployment_id,
737+
deployment.dsc_model_deployment.workflow_req_id,
738+
model_type,
739+
model_name,
740+
),
741+
daemon=True,
739742
)
743+
progress_thread.start()
740744

741745
# we arbitrarily choose last 8 characters of OCID to identify MD in telemetry
742746
telemetry_kwargs = {"ocid": get_ocid_substring(deployment_id, key_len=8)}

0 commit comments

Comments
 (0)