no-op

PiperOrigin-RevId: 548556494

Author: anuartb

tfx/components/infra_validator/model_server_runners/local_docker_runner.py

@@ -14,6 +14,7 @@
 """Module for LocalDockerModelServerRunner."""
 
 import os
+import subprocess
 import time
 from typing import Any, Dict, Optional
 
@@ -65,10 +66,15 @@ def _find_host_port(ports: Dict[str, Any], container_port: int) -> str:
   Raises:
     ValueError: No corresponding host port was found.
   """
+  command = '/sbin/ip route|awk \'/default/ { print $3 }\''
+  ip_address = subprocess.check_output(
+      command, shell=True).decode('utf-8').strip()
+  logging.info('Container IP address: %s', ip_address)
+
   mappings = ports.get('{}/tcp'.format(container_port), [])
   for mapping in mappings:
     if mapping['HostIp'] == '0.0.0.0':
-      return mapping['HostPort']
+      return ip_address, mapping['HostPort']
   else:
     raise ValueError(
         'No HostPort found for ContainerPort={} (all port mappings: {})'
@@ -146,9 +152,11 @@ def WaitUntilRunning(self, deadline):
         continue
       # The container is running :)
       if status == 'running':
-        host_port = _find_host_port(self._container.ports,
-                                    self._serving_binary.container_port)
-        self._endpoint = 'localhost:{}'.format(host_port)
+        ip_address, host_port = _find_host_port(
+            self._container.ports,
+            self._serving_binary.container_port)
+
+        self._endpoint = '{}:{}'.format(ip_address, host_port)
         return
       # Docker status is one of {'created', 'restarting', 'running', 'removing',
       # 'paused', 'exited', or 'dead'}. Status other than 'created' and