Skip to content

Commit 2fed3dc

Browse files
Removing print statements
1 parent 0cf4aa2 commit 2fed3dc

File tree

3 files changed

+7
-6
lines changed

3 files changed

+7
-6
lines changed

ads/aqua/evaluation/evaluation.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1090,9 +1090,6 @@ def download_report(self, eval_id) -> AquaEvalReport:
10901090
content = self._read_from_artifact(
10911091
temp_dir, get_files(temp_dir), EVALUATION_REPORT
10921092
)
1093-
1094-
print("type of content: ", type(content))
1095-
print("content: ", content)
10961093
report = AquaEvalReport(
10971094
evaluation_id=eval_id, content=base64.b64encode(content).decode()
10981095
)

ads/aqua/model/model.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1768,6 +1768,7 @@ def load_readme(self, model_id: str) -> AquaModelReadme:
17681768
content = self.ds_client.get_model_defined_metadatum_artifact_content(
17691769
model_id, AquaModelMetadataKeys.README
17701770
).data.content.decode("utf-8", errors="ignore")
1771+
logger.info(f"Fetched {README} from defined metadata for model: {model_id}")
17711772
except Exception as ex:
17721773
logger.error(
17731774
f"Readme could not be found for model: {model_id} in defined metadata : {str(ex)}"
@@ -1817,6 +1818,9 @@ def load_license(self, model_id: str) -> AquaModelLicense:
18171818
content = self.ds_client.get_model_defined_metadatum_artifact_content(
18181819
model_id, AquaModelMetadataKeys.LICENSE
18191820
).data.content.decode("utf-8", errors="ignore")
1821+
logger.info(
1822+
f"Fetched {LICENSE} from defined metadata for model: {model_id}"
1823+
)
18201824
except Exception as ex:
18211825
logger.error(
18221826
f"License could not be found for model: {model_id} in defined metadata : {str(ex)}"

ads/aqua/modeldeployment/deployment.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -760,9 +760,9 @@ def validate_deployment_params(
760760
model=model, container_family=container_family
761761
)
762762

763-
container_config = self.get_container_config_item(container_family)
764-
container_spec = container_config.spec
765-
cli_params = container_spec.cli_param
763+
container_config = self.get_container_config_item(container_type_key)
764+
container_spec = container_config.spec if container_config else UNKNOWN
765+
cli_params = container_spec.cli_param if container_spec else UNKNOWN
766766

767767
restricted_params = self._find_restricted_params(
768768
cli_params, params, container_type_key

0 commit comments

Comments
 (0)