Skip to content

Commit f8d0962

Browse files
committed
Added validation for fine tuning models.
1 parent 58cb61f commit f8d0962

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

ads/aqua/finetune.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
UpdateModelProvenanceDetails,
1616
)
1717

18+
from ads.aqua import ODSC_MODEL_COMPARTMENT_OCID
1819
from ads.aqua.base import AquaApp
1920
from ads.aqua.data import AquaResourceIdentifier, Resource, Tags
2021
from ads.aqua.exception import AquaFileExistsError, AquaValueError
@@ -192,12 +193,11 @@ def create(
192193
)
193194

194195
source = self.get_source(create_fine_tuning_details.ft_source_id)
195-
# TODO: add the following validation for fine tuning aqua service model. Revisit it when all service models are available
196-
# if source.compartment_id != ODSC_MODEL_COMPARTMENT_OCID:
197-
# raise AquaValueError(
198-
# f"Fine tuning is only supported for Aqua service models in {ODSC_MODEL_COMPARTMENT_OCID}. "
199-
# "Use a valid Aqua service model id instead."
200-
# )
196+
if source.compartment_id != ODSC_MODEL_COMPARTMENT_OCID:
197+
raise AquaValueError(
198+
f"Fine tuning is only supported for Aqua service models in {ODSC_MODEL_COMPARTMENT_OCID}. "
199+
"Use a valid Aqua service model id instead."
200+
)
201201

202202
target_compartment = (
203203
create_fine_tuning_details.compartment_id or COMPARTMENT_OCID

0 commit comments

Comments
 (0)