Skip to content

Commit f722fc1

Browse files
committed
Fixes unit tests
1 parent 70660e9 commit f722fc1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/unitary/with_extras/aqua/test_model.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -481,6 +481,7 @@ def test_get_foundation_models(
481481
}
482482

483483
@patch("ads.aqua.common.utils.query_resource")
484+
@patch("ads.aqua.model.model.get_container_config")
484485
@patch("ads.aqua.model.model.read_file")
485486
@patch.object(DataScienceModel, "from_id")
486487
@patch(
@@ -492,8 +493,8 @@ def test_get_model_fine_tuned(
492493
mock_get_artifact_path,
493494
mock_from_id,
494495
mock_read_file,
495-
mock_query_resource,
496496
mock_get_container_config,
497+
mock_query_resource,
497498
mock_auth,
498499
):
499500
mock_get_container_config.return_value = get_container_config()

0 commit comments

Comments
 (0)