Skip to content

Commit 34a85ee

Browse files
review comments
1 parent 14b683c commit 34a85ee

File tree

4 files changed

+20
-18
lines changed

4 files changed

+20
-18
lines changed

ads/aqua/common/utils.py

Lines changed: 0 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -100,23 +100,6 @@ def get_status(evaluation_status: str, job_run_status: str = None):
100100
JobRun.LIFECYCLE_STATE_NEEDS_ATTENTION: "Missing jobrun information.",
101101
}
102102

103-
CONSOLE_LINK_RESOURCE_TYPE_MAPPING = dict(
104-
datasciencemodel="models",
105-
datasciencemodeldeployment="model-deployments",
106-
datasciencemodeldeploymentdev="model-deployments",
107-
datasciencemodeldeploymentint="model-deployments",
108-
datasciencemodeldeploymentpre="model-deployments",
109-
datasciencejob="jobs",
110-
datasciencejobrun="job-runs",
111-
datasciencejobrundev="job-runs",
112-
datasciencejobrunint="job-runs",
113-
datasciencejobrunpre="job-runs",
114-
datasciencemodelversionset="model-version-sets",
115-
datasciencemodelversionsetpre="model-version-sets",
116-
datasciencemodelversionsetint="model-version-sets",
117-
datasciencemodelversionsetdev="model-version-sets",
118-
)
119-
120103

121104
def random_color_generator(word: str):
122105
seed = sum([ord(c) for c in word]) % 13

ads/aqua/constants.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,3 +41,20 @@
4141
SERVICE_MANAGED_CONTAINER_URI_SCHEME = "dsmc://"
4242
SUPPORTED_FILE_FORMATS = ["jsonl"]
4343
MODEL_BY_REFERENCE_OSS_PATH_KEY = "artifact_location"
44+
45+
CONSOLE_LINK_RESOURCE_TYPE_MAPPING = dict(
46+
datasciencemodel="models",
47+
datasciencemodeldeployment="model-deployments",
48+
datasciencemodeldeploymentdev="model-deployments",
49+
datasciencemodeldeploymentint="model-deployments",
50+
datasciencemodeldeploymentpre="model-deployments",
51+
datasciencejob="jobs",
52+
datasciencejobrun="job-runs",
53+
datasciencejobrundev="job-runs",
54+
datasciencejobrunint="job-runs",
55+
datasciencejobrunpre="job-runs",
56+
datasciencemodelversionset="model-version-sets",
57+
datasciencemodelversionsetpre="model-version-sets",
58+
datasciencemodelversionsetint="model-version-sets",
59+
datasciencemodelversionsetdev="model-version-sets",
60+
)

ads/aqua/evaluation/evaluation.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@
5050
JOB_INFRASTRUCTURE_TYPE_DEFAULT_NETWORKING,
5151
NB_SESSION_IDENTIFIER,
5252
UNKNOWN,
53+
CONSOLE_LINK_RESOURCE_TYPE_MAPPING,
5354
)
5455
from ads.aqua.evaluation.constants import *
5556
from ads.aqua.evaluation.entities import *
@@ -1344,7 +1345,7 @@ def _build_resource_identifier(
13441345
) -> AquaResourceIdentifier:
13451346
"""Constructs AquaResourceIdentifier based on the given ocid and display name."""
13461347
try:
1347-
resource_type = utils.CONSOLE_LINK_RESOURCE_TYPE_MAPPING.get(
1348+
resource_type = CONSOLE_LINK_RESOURCE_TYPE_MAPPING.get(
13481349
utils.get_resource_type(id)
13491350
)
13501351

ads/aqua/model/model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -667,6 +667,7 @@ def _create_model_catalog_entry(
667667
f"Using artifact from service bucket instead of {os_path}"
668668
)
669669

670+
# todo: implement generic copy_folder method
670671
# copy model config from artifact path to user bucket
671672
copy_model_config(
672673
artifact_path=artifact_path, os_path=os_path, auth=self._auth

0 commit comments

Comments
 (0)