Skip to content

Commit ef2c40e

Browse files
Addressing review comments
1 parent e449f0a commit ef2c40e

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

ads/aqua/app.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -313,10 +313,11 @@ def get_config(
313313
f"Base model found for the model: {oci_model.id}. "
314314
f"Loading {config_file_name} for base model {base_model_ocid}."
315315
)
316-
base_model = self.ds_client.get_model(base_model_ocid).data
317-
artifact_path = get_artifact_path(base_model.custom_metadata_list)
318316
if config_folder == ConfigFolder.ARTIFACT:
319317
artifact_path = get_artifact_path(oci_model.custom_metadata_list)
318+
else:
319+
base_model = self.ds_client.get_model(base_model_ocid).data
320+
artifact_path = get_artifact_path(base_model.custom_metadata_list)
320321
else:
321322
logger.info(f"Loading {config_file_name} for model {oci_model.id}...")
322323
artifact_path = get_artifact_path(oci_model.custom_metadata_list)

0 commit comments

Comments
 (0)