Replace mutable default arguments with None.

PiperOrigin-RevId: 597859400
google-deepmind · Jan 12, 2024 · 61ea983 · 61ea983
1 parent 94a2f03
commit 61ea983
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/android_env/components/coordinator.py b/android_env/components/coordinator.py
@@ -45,7 +45,7 @@ def __init__(
       self,
       simulator: base_simulator.BaseSimulator,
       task_manager: task_manager_lib.TaskManager,
-      config: config_classes.CoordinatorConfig = config_classes.CoordinatorConfig(),
+      config: config_classes.CoordinatorConfig | None = None,
   ):
     """Handles communication between AndroidEnv and its components.
 
@@ -55,7 +55,7 @@ def __init__(
     """
     self._simulator = simulator
     self._task_manager = task_manager
-    self._config = config
+    self._config = config or config_classes.CoordinatorConfig()
     self._adb_call_parser: adb_call_parser.AdbCallParser = None
     self._orientation = np.zeros(4, dtype=np.uint8)
     self._interaction_thread: InteractionThread | None = None

diff --git a/android_env/components/task_manager.py b/android_env/components/task_manager.py
@@ -43,7 +43,7 @@ class TaskManager:
   def __init__(
       self,
       task: task_pb2.Task,
-      config: config_classes.TaskManagerConfig = config_classes.TaskManagerConfig(),
+      config: config_classes.TaskManagerConfig | None = None,
   ):
     """Controls task-relevant events and information.
 
@@ -53,7 +53,7 @@ def __init__(
     """
 
     self._task = task
-    self._config = config
+    self._config = config or config_classes.TaskManagerConfig()
     self._lock = threading.Lock()
     self._logcat_thread = None
     self._dumpsys_thread = None