Simplify cell editing logic in DataFrame output and test

Removed unnecessary class expectation when editing cells in OutputDataFrame and simplified flaky retry logic in the related test by using a single wait and class state check. This streamlines cell editing interactions and improves test reliability.

Author: karangattu

shiny/playwright/controller/_output.py

@@ -1105,8 +1105,6 @@ def _edit_cell_no_save(
         self._cell_scroll_if_needed(row=row, col=col, timeout=timeout)
         cell.dblclick(timeout=timeout)
 
-        # Wait for the cell to enter editing mode before filling the textarea
-        expect_to_have_class(cell, "cell-edit-editing", timeout=timeout)
         cell.locator("> textarea").fill(text)
 
     def set_sort(

tests/playwright/shiny/components/data_frame/validate_row_selection_edit_mode/test_validate_row_selection_edit_mode.py

@@ -60,25 +60,15 @@ def test_validate_row_selection_in_edit_mode(
     page.keyboard.press("Escape")
     data_frame._edit_cell_no_save("Temp value", row=1, col=16)
     page.keyboard.press("Escape")
+    # Add a small delay
+    page.wait_for_timeout(100)
     page.keyboard.press("Enter")
-
-    # This interaction (Enter key to start editing) can be flaky, so we retry if needed
-    max_retries = 3
-    for attempt in range(max_retries):
-        try:
-            page.wait_for_timeout(100)
-            data_frame.expect_class_state(
-                "editing",
-                row=1,
-                col=0,
-                timeout=1000,
-            )
-            break
-        except AssertionError:
-            if attempt < max_retries - 1:
-                page.keyboard.press("Enter")
-            else:
-                raise
+    data_frame.expect_class_state(
+        "editing",
+        row=1,
+        col=0,
+        timeout=1000,
+    )
 
     # Click outside the table/Press Escape to exit row focus.
     # Tab to the column name, hit enter. Verify the table becomes sorted.