Hardcode the golden data path. (#4588)

codalab · Jan 27, 2024 · 345c37e · 345c37e
1 parent 128bf4d
commit 345c37e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/ui/ui_tester.py b/tests/ui/ui_tester.py
@@ -306,6 +306,8 @@ def create_path(path):
         return output_dir
 
     def _get_browser_name(self):
+        if self.browser.capabilities['browserName'] == 'chrome-headless-shell':
+            return 'chrome'
         return self.browser.capabilities['browserName']