Skip to content

Commit 5616f90

Browse files
Adding logs for fetching model config
1 parent c16889b commit 5616f90

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

ads/aqua/finetuning/finetuning.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -263,7 +263,7 @@ def create(
263263
compartment_id=target_compartment,
264264
project_id=target_project,
265265
model_by_reference=True,
266-
defined_tags=create_fine_tuning_details.defined_tags
266+
defined_tags=create_fine_tuning_details.defined_tags,
267267
)
268268

269269
ft_job_freeform_tags = {
@@ -594,6 +594,9 @@ def get_finetuning_config(self, model_id: str) -> Dict:
594594
model_id, AquaModelMetadataKeys.FINE_TUNING_CONFIGURATION
595595
)
596596
if config:
597+
logger.info(
598+
f"Fetched {AquaModelMetadataKeys.FINE_TUNING_CONFIGURATION} from defined metadata for model: {model_id}."
599+
)
597600
return config
598601
config = self.get_config(
599602
model_id,

ads/aqua/modeldeployment/deployment.py

+3
Original file line numberDiff line numberDiff line change
@@ -659,6 +659,9 @@ def get_deployment_config(self, model_id: str) -> Dict:
659659
model_id, AquaModelMetadataKeys.DEPLOYMENT_CONFIGURATION
660660
)
661661
if config:
662+
logger.info(
663+
f"Fetched {AquaModelMetadataKeys.DEPLOYMENT_CONFIGURATION} from defined metadata for model: {model_id}."
664+
)
662665
return config
663666
config = self.get_config(
664667
model_id,

0 commit comments

Comments
 (0)