Remove socket.error handling.

We have no code that raises `socket.error` anymore and we should remove this
error handling as it's confusing and irrelevant. This is a relic from the time
AndroidEnv used to interact with the Android Emulator via telnet.

PiperOrigin-RevId: 730579911

Author: kenjitoyama

android_env/components/action_fns.py

@@ -15,8 +15,6 @@
 
 """Functions to convert actions between different components' formats."""
 
-import socket
-
 from absl import logging
 from android_env.components import action_type as action_type_lib
 from android_env.components import errors
@@ -61,7 +59,7 @@ def send_action_to_simulator(
         simulator.send_key(action['keycode'].item(0), event_type='keyup')
       case action_type_lib.ActionType.KEYPRESS:
         simulator.send_key(action['keycode'].item(0), event_type='keypress')
-  except (socket.error, errors.SendActionError):
+  except errors.SendActionError:
     logging.exception('Unable to execute action: %r', action)
     return False
 

android_env/components/action_fns_test.py

@@ -13,7 +13,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import socket
 from unittest import mock
 
 from absl.testing import absltest
@@ -45,30 +44,6 @@ def test_send_action_to_simulator_missing_action_type(self):
         1,
     )
 
-  def test_send_action_to_simulator_socket_error(self):
-    """Returns `False` if the simulator raises a socket error."""
-
-    # Arrange.
-    simulator = mock.create_autospec(base_simulator.BaseSimulator)
-    simulator.send_touch.side_effect = socket.error('not today')
-    action = {
-        'action_type': action_type_lib.ActionType.TOUCH,
-        'touch_position': np.array([0.3, 0.5], np.float32),
-    }
-
-    # Act.
-    output = action_fns.send_action_to_simulator(
-        action,
-        simulator,
-        800,
-        600,
-        1,
-    )
-
-    # Assert.
-    self.assertFalse(output)
-    simulator.send_touch.assert_called_once()
-
   def test_send_action_to_simulator_sendactionerror(self):
     """Returns `False` if the simulator raises a SendActionError."""
 

android_env/components/coordinator.py

@@ -16,7 +16,6 @@
 """Coordinator handles interaction between internal components of AndroidEnv."""
 
 import copy
-import socket
 import time
 from typing import Any
 
@@ -238,7 +237,7 @@ def rl_step(self, agent_action: dict[str, np.ndarray]) -> dm_env.TimeStep:
     # Get data from the simulator.
     try:
       simulator_signals = self._gather_simulator_signals()
-    except (errors.ReadObservationError, socket.error):
+    except errors.ReadObservationError:
       logging.exception('Unable to fetch observation. Restarting simulator.')
       self._stats['relaunch_count_fetch_observation'] += 1
       self._simulator_healthy = False