@@ -1438,11 +1438,11 @@ def test_verify_compatibility(self):
1438
1438
@patch ("ads.model.deployment.model_deployment.ModelDeployment.deploy" )
1439
1439
@patch .object (AquaApp , "get_container_config" )
1440
1440
@patch (
1441
- "ads.aqua.modeldeployment.entities.CreateModelDeploymentDetails.validate_ft_model_v2 "
1441
+ "ads.aqua.modeldeployment.entities.CreateModelDeploymentDetails.validate_base_model "
1442
1442
)
1443
1443
def test_create_deployment_for_foundation_model (
1444
1444
self ,
1445
- mock_validate_ft_model_v2 ,
1445
+ mock_validate_base_model ,
1446
1446
mock_get_container_config ,
1447
1447
mock_deploy ,
1448
1448
mock_get_container_image ,
@@ -1518,7 +1518,7 @@ def test_create_deployment_for_foundation_model(
1518
1518
defined_tags = defined_tags ,
1519
1519
)
1520
1520
1521
- mock_validate_ft_model_v2 .assert_called ()
1521
+ mock_validate_base_model .assert_called ()
1522
1522
mock_create .assert_called_with (
1523
1523
model = TestDataset .MODEL_ID ,
1524
1524
compartment_id = TestDataset .USER_COMPARTMENT_ID ,
@@ -1544,11 +1544,11 @@ def test_create_deployment_for_foundation_model(
1544
1544
@patch ("ads.model.deployment.model_deployment.ModelDeployment.deploy" )
1545
1545
@patch .object (AquaApp , "get_container_config" )
1546
1546
@patch (
1547
- "ads.aqua.modeldeployment.entities.CreateModelDeploymentDetails.validate_ft_model_v2 "
1547
+ "ads.aqua.modeldeployment.entities.CreateModelDeploymentDetails.validate_base_model "
1548
1548
)
1549
1549
def test_create_deployment_for_fine_tuned_model (
1550
1550
self ,
1551
- mock_validate_ft_model_v2 ,
1551
+ mock_validate_base_model ,
1552
1552
mock_get_container_config ,
1553
1553
mock_deploy ,
1554
1554
mock_get_container_image ,
@@ -1619,7 +1619,7 @@ def test_create_deployment_for_fine_tuned_model(
1619
1619
predict_log_id = "ocid1.log.oc1.<region>.<OCID>" ,
1620
1620
)
1621
1621
1622
- mock_validate_ft_model_v2 .assert_called ()
1622
+ mock_validate_base_model .assert_called ()
1623
1623
mock_create .assert_called_with (
1624
1624
model = TestDataset .MODEL_ID ,
1625
1625
compartment_id = TestDataset .USER_COMPARTMENT_ID ,
@@ -1643,11 +1643,11 @@ def test_create_deployment_for_fine_tuned_model(
1643
1643
@patch ("ads.model.deployment.model_deployment.ModelDeployment.deploy" )
1644
1644
@patch .object (AquaApp , "get_container_config" )
1645
1645
@patch (
1646
- "ads.aqua.modeldeployment.entities.CreateModelDeploymentDetails.validate_ft_model_v2 "
1646
+ "ads.aqua.modeldeployment.entities.CreateModelDeploymentDetails.validate_base_model "
1647
1647
)
1648
1648
def test_create_deployment_for_gguf_model (
1649
1649
self ,
1650
- mock_validate_ft_model_v2 ,
1650
+ mock_validate_base_model ,
1651
1651
mock_get_container_config ,
1652
1652
mock_deploy ,
1653
1653
mock_get_container_image ,
@@ -1720,7 +1720,7 @@ def test_create_deployment_for_gguf_model(
1720
1720
memory_in_gbs = 60.0 ,
1721
1721
)
1722
1722
1723
- mock_validate_ft_model_v2 .assert_called ()
1723
+ mock_validate_base_model .assert_called ()
1724
1724
mock_create .assert_called_with (
1725
1725
model = TestDataset .MODEL_ID ,
1726
1726
compartment_id = TestDataset .USER_COMPARTMENT_ID ,
@@ -1748,11 +1748,11 @@ def test_create_deployment_for_gguf_model(
1748
1748
@patch ("ads.model.deployment.model_deployment.ModelDeployment.deploy" )
1749
1749
@patch .object (AquaApp , "get_container_config" )
1750
1750
@patch (
1751
- "ads.aqua.modeldeployment.entities.CreateModelDeploymentDetails.validate_ft_model_v2 "
1751
+ "ads.aqua.modeldeployment.entities.CreateModelDeploymentDetails.validate_base_model "
1752
1752
)
1753
1753
def test_create_deployment_for_tei_byoc_embedding_model (
1754
1754
self ,
1755
- mock_validate_ft_model_v2 ,
1755
+ mock_validate_base_model ,
1756
1756
mock_get_container_config ,
1757
1757
mock_deploy ,
1758
1758
mock_get_container_image ,
@@ -1828,7 +1828,7 @@ def test_create_deployment_for_tei_byoc_embedding_model(
1828
1828
cmd_var = [],
1829
1829
)
1830
1830
1831
- mock_validate_ft_model_v2 .assert_called ()
1831
+ mock_validate_base_model .assert_called ()
1832
1832
mock_create .assert_called_with (
1833
1833
model = TestDataset .MODEL_ID ,
1834
1834
compartment_id = TestDataset .USER_COMPARTMENT_ID ,
0 commit comments