Change accessibility_node_to_ui_element() function from internal to p…

…ublic. PiperOrigin-RevId: 679673979
google-research · Sep 27, 2024 · 8bc0c84 · 8bc0c84
1 parent 899d4d6
commit 8bc0c84
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/android_world/env/representation_utils.py b/android_world/env/representation_utils.py
@@ -76,7 +76,7 @@ class UIElement:
   resource_id: Optional[str] = None
 
 
-def _accessibility_node_to_ui_element(
+def accessibility_node_to_ui_element(
     node: Any,
     screen_size: Optional[tuple[int, int]] = None,
 ) -> UIElement:
@@ -158,7 +158,7 @@ def forest_to_ui_elements(
         if exclude_invisible_elements and not node.is_visible_to_user:
           continue
         else:
-          elements.append(_accessibility_node_to_ui_element(node, screen_size))
+          elements.append(accessibility_node_to_ui_element(node, screen_size))
   return elements
 
 

diff --git a/android_world/env/representation_utils_test.py b/android_world/env/representation_utils_test.py
@@ -100,7 +100,7 @@ def test_normalize_bboxes(
     node = mock.MagicMock()
     node.bounds_in_screen = node_bounds
 
-    ui_element = representation_utils._accessibility_node_to_ui_element(
+    ui_element = representation_utils.accessibility_node_to_ui_element(
         node, screen_size
     )
     self.assertEqual(ui_element.bbox_pixels.x_min, node_bounds.left)