fix: Use the correct dockerhub image tag when building feature servers

sdk/python/feast/infra/aws.py

@@ -119,6 +119,8 @@ def _deploy_feature_server(self, project: str, image_uri: str):
         lambda_client = boto3.client("lambda")
         api_gateway_client = boto3.client("apigatewayv2")
         function = aws_utils.get_lambda_function(lambda_client, resource_name)
+        _logger.debug("Using function name: %s", resource_name)
+        _logger.debug("Found function: %s", function)
 
         if function is None:
             # If the Lambda function does not exist, create it.
@@ -262,7 +264,9 @@ def _upload_docker_image(
         _logger.info(
             f"Pulling remote image {Style.BRIGHT + Fore.GREEN}{dockerhub_image}{Style.RESET_ALL}"
         )
-        for line in docker_client.api.pull(dockerhub_image, stream=True, decode=True):
+        for line in docker_client.api.pull(
+            dockerhub_image.replace("_", "."), stream=True, decode=True
+        ):
             _logger.debug(f"  {line}")
 
         auth_token = ecr_client.get_authorization_token()["authorizationData"][0][
@@ -279,7 +283,7 @@ def _upload_docker_image(
         )
         _logger.debug(f"  {login_status}")
 
-        image = docker_client.images.get(dockerhub_image)
+        image = docker_client.images.get(dockerhub_image.replace("_", "."))
         image_remote_name = f"{repository_uri}:{docker_image_version}"
         _logger.info(
             f"Pushing local image to remote {Style.BRIGHT + Fore.GREEN}{image_remote_name}{Style.RESET_ALL}"

sdk/python/tests/integration/online_store/test_universal_online.py

@@ -1,5 +1,6 @@
 import datetime
 import itertools
+import logging
 import os
 import time
 import unittest
@@ -35,6 +36,8 @@
 )
 from tests.utils.data_source_utils import prep_file_source
 
+_logger = logging.getLogger(__name__)
+
 
 @pytest.mark.integration
 def test_entity_ttl_online_store(local_redis_environment, redis_universal_data_sources):
@@ -262,16 +265,17 @@ def _get_online_features_dict_remotely(
         request["features"] = features
     else:
         request["feature_service"] = features.name
-    for _ in range(25):
+    for attempt in range(25):
         # Send the request to the remote feature server and get the response in JSON format
         response = requests.post(
             f"{endpoint}/get-online-features", json=request, timeout=30
         ).json()
+        _logger.info(f"Attempt: {attempt}, {response}")
         # Retry if the response is internal server error, which can happen when lambda is being restarted
         if response.get("message") != "Internal Server Error":
             break
         # Sleep between retries to give the server some time to start
-        time.sleep(1)
+        time.sleep(15)
     else:
         raise Exception("Failed to get online features from remote feature server")
     if "metadata" not in response: