Skip to content

Commit df5037e

Browse files
Addressing review comments
1 parent 10ab6de commit df5037e

File tree

3 files changed

+2
-5
lines changed

3 files changed

+2
-5
lines changed

ads/aqua/extension/common_ws_msg_handler.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@ def __init__(self, message: Union[str, bytes]):
2323
@handle_exceptions
2424
def process(self) -> AdsVersionResponse | CompatibilityCheckResponse:
2525
request = json.loads(self.message)
26-
print("request: {}".format(request))
2726
if request.get('kind') == 'AdsVersion':
2827
version = metadata.version("oracle_ads")
2928
response = AdsVersionResponse(

ads/aqua/extension/evaluation_ws_msg_handler.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,8 +36,7 @@ def process(self) -> ListEvaluationsResponse | EvaluationDetailsResponse:
3636
def list_evaluations(request) -> ListEvaluationsResponse:
3737

3838
eval_list = AquaEvaluationApp().list(
39-
request.get("compartment_id") or COMPARTMENT_OCID,
40-
request.get("project_id"),
39+
request.get("compartment_id") or COMPARTMENT_OCID
4140
)
4241
response = ListEvaluationsResponse(
4342
message_id=request["message_id"],

ads/aqua/extension/models_ws_msg_handler.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
from ads.aqua.extension.models.ws_models import RequestResponseType, ListModelsResponse, ListModelsRequest, \
77
ModelDetailsResponse
88
from ads.aqua.model import AquaModelApp
9-
from ads.config import COMPARTMENT_OCID
109

1110

1211
class AquaModelWSMsgHandler(AquaWSMsgHandler):
@@ -23,7 +22,7 @@ def process(self) -> ListModelsResponse | ModelDetailsResponse:
2322
request = json.loads(self.message)
2423
if request.get('kind') == 'ListModels':
2524
models_list = AquaModelApp().list(
26-
compartment_id=request.get("compartment_id") or COMPARTMENT_OCID,
25+
compartment_id=request.get("compartment_id"),
2726
project_id=request.get("project_id"),
2827
model_type=request.get("model_type")
2928
)

0 commit comments

Comments
 (0)