minimal cleanup

Author: patricebechard

browsergym/core/src/browsergym/core/hint_labeling.py

@@ -1,19 +1,18 @@
+import json
+import logging
 from importlib import resources
+from queue import Queue
 from typing import Dict, List, Optional
 
-from queue import Queue, Empty
 import playwright.sync_api
 from pydantic import BaseModel, Field
-import logging
-import json
+
 from . import _get_global_playwright, hint_labeling_files
 
 logger = logging.getLogger(__name__)
 
 HINT_LABELING_DIR = resources.files(hint_labeling_files)
 
-# ------- Data Classes -------
-
 
 class HintLabelingInputs(BaseModel):
     goal: str
@@ -22,13 +21,9 @@ class HintLabelingInputs(BaseModel):
     axtree: str
     history: List[Dict[str, str]] = Field(default_factory=list)
     hint: str = ""
-    # keep 'suggestions' on Python side, but we’ll map to UI 'suggestions'
     suggestions: List[Dict[str, str]] = Field(default_factory=list)
 
 
-# ------- Hint Labeling backend class -------
-
-
 class HintLabeling:
     def __init__(self, headless: bool, window_size=(600, 1000), *args, **kwargs):
 

browsergym/core/src/browsergym/utils/obs.py

@@ -320,13 +320,13 @@ def dfs(node_idx: int, depth: int, parent_node_filtered: bool, parent_node_name:
             pass
         else:
             node_name = node["name"]["value"]
-            # if node_name == "":
-            #     try:
-            #         node_name_sources = node["name"]["sources"]
-            #         node_name_sources = [elem for elem in node_name_sources if elem.get("type") == "contents"][0]
-            #         node_name = node_name_sources.get("value", {}).get("value", "")
-            #     except Exception as e:
-            #         logger.warning(f"Failed to extract node name: {e}")
+            if node_name == "":
+                try:
+                    node_name_sources = node["name"]["sources"]
+                    node_name_sources = [elem for elem in node_name_sources if elem.get("type") == "contents"][0]
+                    node_name = node_name_sources.get("value", {}).get("value", "")
+                except Exception as e:
+                    logger.warning(f"Failed to extract node name: {e}")
             if "value" in node and "value" in node["value"]:
                 node_value = node["value"]["value"]
             else: