@@ -1438,11 +1438,11 @@ def test_verify_compatibility(self):
14381438 @patch ("ads.model.deployment.model_deployment.ModelDeployment.deploy" )
14391439 @patch .object (AquaApp , "get_container_config" )
14401440 @patch (
1441- "ads.aqua.modeldeployment.entities.CreateModelDeploymentDetails.validate_ft_model_v2 "
1441+ "ads.aqua.modeldeployment.entities.CreateModelDeploymentDetails.validate_base_model "
14421442 )
14431443 def test_create_deployment_for_foundation_model (
14441444 self ,
1445- mock_validate_ft_model_v2 ,
1445+ mock_validate_base_model ,
14461446 mock_get_container_config ,
14471447 mock_deploy ,
14481448 mock_get_container_image ,
@@ -1518,7 +1518,7 @@ def test_create_deployment_for_foundation_model(
15181518 defined_tags = defined_tags ,
15191519 )
15201520
1521- mock_validate_ft_model_v2 .assert_called ()
1521+ mock_validate_base_model .assert_called ()
15221522 mock_create .assert_called_with (
15231523 model = TestDataset .MODEL_ID ,
15241524 compartment_id = TestDataset .USER_COMPARTMENT_ID ,
@@ -1544,11 +1544,11 @@ def test_create_deployment_for_foundation_model(
15441544 @patch ("ads.model.deployment.model_deployment.ModelDeployment.deploy" )
15451545 @patch .object (AquaApp , "get_container_config" )
15461546 @patch (
1547- "ads.aqua.modeldeployment.entities.CreateModelDeploymentDetails.validate_ft_model_v2 "
1547+ "ads.aqua.modeldeployment.entities.CreateModelDeploymentDetails.validate_base_model "
15481548 )
15491549 def test_create_deployment_for_fine_tuned_model (
15501550 self ,
1551- mock_validate_ft_model_v2 ,
1551+ mock_validate_base_model ,
15521552 mock_get_container_config ,
15531553 mock_deploy ,
15541554 mock_get_container_image ,
@@ -1619,7 +1619,7 @@ def test_create_deployment_for_fine_tuned_model(
16191619 predict_log_id = "ocid1.log.oc1.<region>.<OCID>" ,
16201620 )
16211621
1622- mock_validate_ft_model_v2 .assert_called ()
1622+ mock_validate_base_model .assert_called ()
16231623 mock_create .assert_called_with (
16241624 model = TestDataset .MODEL_ID ,
16251625 compartment_id = TestDataset .USER_COMPARTMENT_ID ,
@@ -1643,11 +1643,11 @@ def test_create_deployment_for_fine_tuned_model(
16431643 @patch ("ads.model.deployment.model_deployment.ModelDeployment.deploy" )
16441644 @patch .object (AquaApp , "get_container_config" )
16451645 @patch (
1646- "ads.aqua.modeldeployment.entities.CreateModelDeploymentDetails.validate_ft_model_v2 "
1646+ "ads.aqua.modeldeployment.entities.CreateModelDeploymentDetails.validate_base_model "
16471647 )
16481648 def test_create_deployment_for_gguf_model (
16491649 self ,
1650- mock_validate_ft_model_v2 ,
1650+ mock_validate_base_model ,
16511651 mock_get_container_config ,
16521652 mock_deploy ,
16531653 mock_get_container_image ,
@@ -1720,7 +1720,7 @@ def test_create_deployment_for_gguf_model(
17201720 memory_in_gbs = 60.0 ,
17211721 )
17221722
1723- mock_validate_ft_model_v2 .assert_called ()
1723+ mock_validate_base_model .assert_called ()
17241724 mock_create .assert_called_with (
17251725 model = TestDataset .MODEL_ID ,
17261726 compartment_id = TestDataset .USER_COMPARTMENT_ID ,
@@ -1748,11 +1748,11 @@ def test_create_deployment_for_gguf_model(
17481748 @patch ("ads.model.deployment.model_deployment.ModelDeployment.deploy" )
17491749 @patch .object (AquaApp , "get_container_config" )
17501750 @patch (
1751- "ads.aqua.modeldeployment.entities.CreateModelDeploymentDetails.validate_ft_model_v2 "
1751+ "ads.aqua.modeldeployment.entities.CreateModelDeploymentDetails.validate_base_model "
17521752 )
17531753 def test_create_deployment_for_tei_byoc_embedding_model (
17541754 self ,
1755- mock_validate_ft_model_v2 ,
1755+ mock_validate_base_model ,
17561756 mock_get_container_config ,
17571757 mock_deploy ,
17581758 mock_get_container_image ,
@@ -1828,7 +1828,7 @@ def test_create_deployment_for_tei_byoc_embedding_model(
18281828 cmd_var = [],
18291829 )
18301830
1831- mock_validate_ft_model_v2 .assert_called ()
1831+ mock_validate_base_model .assert_called ()
18321832 mock_create .assert_called_with (
18331833 model = TestDataset .MODEL_ID ,
18341834 compartment_id = TestDataset .USER_COMPARTMENT_ID ,
0 commit comments