fix test

Signed-off-by: sirutBuasai <sirutbuasai27@outlook.com>

Author: sirutBuasai

test/docs/test_image_config.py

@@ -20,7 +20,7 @@
     ImageConfig,
     build_image_row,
     check_public_registry,
-    get_latest_image,
+    get_latest_image_uri,
     load_legacy_images,
     load_repository_images,
     sort_by_version,
@@ -278,12 +278,12 @@ def test_mixed(self, caplog):
 
 class TestGetLatestImage:
     def test_success(self, mock_paths):
-        LOGGER.debug("Testing get_latest_image for valid repo/platform")
-        uri = get_latest_image("mock-repo", "ec2")
+        LOGGER.debug("Testing get_latest_image_uri for valid repo/platform")
+        uri = get_latest_image_uri("mock-repo", "ec2")
         LOGGER.debug(f"Latest image URI: {uri}")
         assert "mock-repo" in uri
         assert "2.0.0-gpu-py312" in uri
-        LOGGER.info("get_latest_image test passed")
+        LOGGER.info("get_latest_image_uri test passed")
 
     @pytest.mark.parametrize(
         "repo,platform",
@@ -292,4 +292,4 @@ def test_success(self, mock_paths):
     )
     def test_not_found(self, mock_paths, repo, platform):
         with pytest.raises(ValueError, match="Image not found"):
-            get_latest_image(repo, platform)
+            get_latest_image_uri(repo, platform)