From 5ea7d37d2b47d7d98a1f19f8c59760b6a86a483b Mon Sep 17 00:00:00 2001
From: Panupong Pasupat <ppasupat@stanford.edu>
Date: Sat, 22 Oct 2022 22:52:28 -0700
Subject: [PATCH 1/2] Make the code work with Python 3 + Selenium

---
 html/miniwob/click-dialog-2.html |   4 +-
 python/README.md                 |   2 +-
 python/miniwob/action.py         |  16 +++--
 python/miniwob/environment.py    |   6 +-
 python/miniwob/fields.py         |  18 ++---
 python/miniwob/instance.py       |  18 ++---
 python/miniwob/screenshot.py     |   8 +--
 python/miniwob/state.py          |   8 +--
 python/miniwob/utils.py          |  10 +--
 python/requirements.txt          |   5 +-
 python/test/test_action.py       |  32 ++++-----
 python/test/test_environment.py  | 120 +++++++++++++++----------------
 record.py                        |   8 +--
 13 files changed, 127 insertions(+), 128 deletions(-)

diff --git a/html/miniwob/click-dialog-2.html b/html/miniwob/click-dialog-2.html
index 2dd3af0e..e5dfe32e 100644
--- a/html/miniwob/click-dialog-2.html
+++ b/html/miniwob/click-dialog-2.html
@@ -41,11 +41,11 @@
     position: {my: 'center', at: 'center', of: document.getElementById('area')},
     buttons: [
       { text: 'Cancel', click: function(e) {
-        var r = e.toElement.innerHTML === expectedButton ? 1.0 : -1.0;
+        var r = e.target.innerHTML === expectedButton ? 1.0 : -1.0;
         core.endEpisode(r, r > 0);
       } },
       { text: 'OK', click: function(e) {
-        var r = e.toElement.innerHTML === expectedButton ? 1.0 : -1.0;
+        var r = e.target.innerHTML === expectedButton ? 1.0 : -1.0;
         core.endEpisode(r, r > 0);
       } }
     ]
diff --git a/python/README.md b/python/README.md
index 8eaf056f..77140602 100644
--- a/python/README.md
+++ b/python/README.md
@@ -10,7 +10,7 @@
 
 - Selenium
   - Outside this repository, download
-    [ChromeDriver](https://sites.google.com/a/chromium.org/chromedriver/downloads).
+    [ChromeDriver](https://chromedriver.chromium.org/downloads).
     Unzip it and then add the directory
     containing the `chromedriver` executable to the `PATH` environment variable
     ```
diff --git a/python/miniwob/action.py b/python/miniwob/action.py
index 1cc131af..22eaae63 100644
--- a/python/miniwob/action.py
+++ b/python/miniwob/action.py
@@ -3,11 +3,11 @@
 import logging
 
 from selenium.webdriver.common.action_chains import ActionChains
+from selenium.webdriver.common.by import By
 
 
-class MiniWoBAction(object):
+class MiniWoBAction(metaclass=abc.ABCMeta):
     """Defines an action in its __call__ method."""
-    __metaclass__ = abc.ABCMeta
 
     @abc.abstractmethod
     def __call__(self, driver):
@@ -61,10 +61,12 @@ def __init__(self, left, top):
 
     def __call__(self, driver):
         """Clicks at coordinates (left, top)"""
-        body = driver.find_element_by_tag_name('body')
+        body = driver.find_element(By.TAG_NAME, 'body')
+        # The offset is from the center, not top-left.
+        x = - body.size['width'] / 2 + self.left
+        y = - body.size['height'] / 2 + self.top
         chain = ActionChains(driver)
-        chain.move_to_element_with_offset(
-                body, self.left, self.top).click().perform()
+        chain.move_to_element_with_offset(body, x, y).click().perform()
 
     @property
     def left(self):
@@ -114,7 +116,7 @@ def __init__(self, element, fail_hard=False):
     def __call__(self, driver):
         if self.element.tag == 'select':
             # SPECIAL CASE: <select>
-            body = driver.find_element_by_tag_name('body')
+            body = driver.find_element(By.TAG_NAME, 'body')
             chain = ActionChains(driver)
             chain.move_to_element_with_offset(
                     body, self.element.left + 5,
@@ -126,7 +128,7 @@ def __call__(self, driver):
             if self._fail_hard:
                 raise RuntimeError('{} failed: {}'.format(self, result))
             else:
-                logging.warn('%s failed: %s', self, result)
+                logging.warning('%s failed: %s', self, result)
 
     @property
     def ref(self):
diff --git a/python/miniwob/environment.py b/python/miniwob/environment.py
index c507b6cf..636282f6 100644
--- a/python/miniwob/environment.py
+++ b/python/miniwob/environment.py
@@ -56,7 +56,7 @@ def configure(self, num_instances=1, seeds=None, **kwargs):
         for instance in self.instances:
             instance.close()
         self.instances = []
-        for index in xrange(num_instances):
+        for index in range(num_instances):
             logging.info('Starting WebDriver Instance %d', index)
             instance = MiniWoBInstance(
                     index, self.subdomain, seeds[index], **kwargs)
@@ -176,13 +176,13 @@ def test_environment():
     try:
         task_name = sys.argv[1]
     except IndexError:
-        print 'Usage: python {} TASK_NAME'.format(sys.argv[0])
+        print('Usage: python {} TASK_NAME'.format(sys.argv[0]))
         exit(1)
     env = MiniWoBEnvironment(task_name)
     base_url = os.environ.get('MINIWOB_BASE_URL')
     env.configure(num_instances=1, seeds=[0], base_url=base_url)
     states = env.reset()
-    print states[0].dom.visualize()
+    print(states[0].dom.visualize())
     env.close()
 
 if __name__ == '__main__':
diff --git a/python/miniwob/fields.py b/python/miniwob/fields.py
index 9d6d6020..c29d8790 100644
--- a/python/miniwob/fields.py
+++ b/python/miniwob/fields.py
@@ -45,14 +45,14 @@ def __len__(self):
 
     @property
     def keys(self):
-        return self._d.keys()
+        return list(self._d.keys())
 
     @property
     def values(self):
-        return self._d.values()
+        return list(self._d.values())
 
     def __repr__(self):
-        return '\n'.join(u'{}: {}'.format(k, repr(v)) for k, v in self._d.items())
+        return '\n'.join('{}: {}'.format(k, repr(v)) for k, v in list(self._d.items()))
     __str__ = __repr__
 
 
@@ -134,7 +134,7 @@ def extract_click_checkboxes(utterance):
     else:
         targets = re.split(', ?', targets)
     fields = dict(zip(
-        ["target {}".format(i) for i in xrange(len(targets))], targets))
+        ["target {}".format(i) for i in range(len(targets))], targets))
     fields["button"] = "submit"
     return Fields(fields)
 FIELD_EXTRACTORS['click-checkboxes'] = extract_click_checkboxes
@@ -156,7 +156,7 @@ def extract_click_checkboxes_soft(utterance):
     targets = re.match(r'Select words similar to (.*) and click Submit\.', utterance).group(1)
     targets = re.split(', ?', targets)
     fields = dict(zip(
-        ["target {}".format(i) for i in xrange(len(targets))], targets))
+        ["target {}".format(i) for i in range(len(targets))], targets))
     fields["button"] = "submit"
     return Fields(fields)
 FIELD_EXTRACTORS['click-checkboxes-soft'] = extract_click_checkboxes_soft
@@ -900,17 +900,17 @@ def extract_utterances():
     try:
         task_name = sys.argv[1]
     except:
-        print >> sys.stderr, 'Usage: {} task_name'.format(sys.argv[0])
+        print('Usage: {} task_name'.format(sys.argv[0]), file=sys.stderr)
         exit(1)
     FIELD_EXTRACTORS[task_name] = lambda utt: Fields({})
     from miniwob.environment import MiniWoBEnvironment
     env = MiniWoBEnvironment(task_name)
     base_url = os.environ.get('MINIWOB_BASE_URL')
-    env.configure(num_instances=4, seeds=range(4), base_url=base_url)
-    for i in xrange(25):
+    env.configure(num_instances=4, seeds=list(range(4)), base_url=base_url)
+    for i in range(25):
         states = env.reset()
         for state in states:
-            print 'UTT:\t{}'.format(state.utterance.replace('\n', ' '))
+            print('UTT:\t{}'.format(state.utterance.replace('\n', ' ')))
     env.close()
 
 if __name__ == '__main__':
diff --git a/python/miniwob/instance.py b/python/miniwob/instance.py
index 6bbfa8de..234c9af9 100644
--- a/python/miniwob/instance.py
+++ b/python/miniwob/instance.py
@@ -1,9 +1,9 @@
 import json
 import logging
-from Queue import Queue
+from queue import Queue
 import time
 import traceback
-import urlparse
+import urllib.parse
 from threading import Thread
 
 import numpy as np
@@ -76,14 +76,14 @@ def __init__(self, index, subdomain, seed, headless=False,
             assert not base_url.startswith('file://'),\
                     ('For {} domain, MINIWOB_BASE_URL cannot be file://. '
                      ' See "Run a simple server" in README').format(subdomain)
-            self.url = urlparse.urljoin(base_url,
+            self.url = urllib.parse.urljoin(base_url,
                     subdomain.replace('.', '/') + '/wrapper.html')
             self.window_width = self.FLIGHT_WINDOW_WIDTH
             self.window_height = self.FLIGHT_WINDOW_HEIGHT
             self.task_width = self.FLIGHT_TASK_WIDTH
             self.task_height = self.FLIGHT_TASK_HEIGHT
         else:
-            self.url = urlparse.urljoin(base_url,
+            self.url = urllib.parse.urljoin(base_url,
                     'miniwob/{}.html'.format(subdomain))
             self.window_width = self.WINDOW_WIDTH
             self.window_height = self.WINDOW_HEIGHT
@@ -156,7 +156,7 @@ def create_driver(self):
                     .format(self.window_width, self.window_height))
             options.add_argument('window-position={},{}'
                     .format(9000, 30 + self.index * (self.window_height + 30)))
-        self.driver = webdriver.Chrome(chrome_options=options)
+        self.driver = webdriver.Chrome(options=options)
         self.driver.implicitly_wait(5)
         if self.headless:
             self.driver.get(self.url)
@@ -247,13 +247,9 @@ def begin_task(self, seed=None):
         if seed is not None:
             self.set_seed(seed)
         self.set_mode(self.mode)
-        # Wait for the sync screen, then click on it
-        #element = WebDriverWait(self.driver, 5).until(
-        #                EC.element_to_be_clickable((By.ID, self.SYNC_SCREEN_ID)))
-        #self.driver.find_element_by_id(self.SYNC_SCREEN_ID).click()
         self.driver.execute_script('core.startEpisodeReal();')
         if self.block_on_reset:
-            for _ in xrange(self.RESET_BLOCK_MAX_ATTEMPT):
+            for _ in range(self.RESET_BLOCK_MAX_ATTEMPT):
                 if self.driver.execute_script('return WOB_TASK_READY;'):
                     break
                 time.sleep(self.RESET_BLOCK_SLEEP_TIME)
@@ -274,7 +270,7 @@ def perform(self, action):
         """
         if action is not None:
             if self.get_metadata()['done']:
-                logging.warn('Cannot call %s on instance %d, which is already done',
+                logging.warning('Cannot call %s on instance %d, which is already done',
                         action, self.index)
             else:
                 action(self.driver)
diff --git a/python/miniwob/screenshot.py b/python/miniwob/screenshot.py
index 381820b3..87e3585a 100644
--- a/python/miniwob/screenshot.py
+++ b/python/miniwob/screenshot.py
@@ -4,7 +4,7 @@
 
 import numpy as np
 
-from cStringIO import StringIO
+from io import StringIO
 from PIL import Image, ImageDraw
 
 
@@ -79,8 +79,8 @@ def create_gif(path_prefix):
 
 if __name__ == '__main__':
     if len(sys.argv) != 2:
-        print 'Usage: {} PATH_PREFIX'.format(sys.argv[0])
-        print '  where PATH_PREFIX is something like '\
-                'data/experiments/123_unnamed/traces/test/2000-img/2000-3'
+        print('Usage: {} PATH_PREFIX'.format(sys.argv[0]))
+        print('  where PATH_PREFIX is something like '
+              'data/experiments/123_unnamed/traces/test/2000-img/2000-3')
         exit(1)
     create_gif(sys.argv[1])
diff --git a/python/miniwob/state.py b/python/miniwob/state.py
index 5091eb18..a78c4730 100644
--- a/python/miniwob/state.py
+++ b/python/miniwob/state.py
@@ -24,7 +24,7 @@ def __init__(self, utterance, fields, dom_info):
         """
         ################
         # Parse utterance
-        assert isinstance(utterance, basestring)
+        assert isinstance(utterance, str)
         self._phrase = Phrase(utterance)
         self._fields = fields
         ################
@@ -145,7 +145,7 @@ def __init__(self, raw_dom, parent=None, dom_elements=None):
         if self.tag == 't':
             self._ref = None  # ignore refs for text, since they are unreliable
         if 'text' in raw_dom:
-            self._text = unicode(raw_dom['text'])
+            self._text = str(raw_dom['text'])
         else:
             self._text = None
         self._value = raw_dom.get('value')
@@ -168,7 +168,7 @@ def __init__(self, raw_dom, parent=None, dom_elements=None):
         # Fix a bug where sometimes children are created even though all children are <t>
         # (which will incorrectly make this element a non-leaf and thus unclickable)
         if self._children and all(child.tag == 't' for child in self._children):
-            self._text = u' '.join(child.text for child in self._children)
+            self._text = ' '.join(child.text for child in self._children)
             self._children = []
         # Add to the collection
         if dom_elements is not None:
@@ -350,7 +350,7 @@ def visualize(self, join=True):
         lines = []
         lines.append('- {}'.format(self))
         for i, child in enumerate(self.children):
-            if isinstance(child, unicode):
+            if isinstance(child, str):
                 child = child[:20] + '...' if len(child) > 20 else child
                 lines.append('  |- "{}"'.format(child))
             else:
diff --git a/python/miniwob/utils.py b/python/miniwob/utils.py
index 45783533..0ff8a246 100644
--- a/python/miniwob/utils.py
+++ b/python/miniwob/utils.py
@@ -1,4 +1,4 @@
-import regex as re
+import re
 
 
 def strip_punctuation(uni):
@@ -10,7 +10,7 @@ def strip_punctuation(uni):
     Returns:
         unicode
     """
-    return re.sub(ur"\p{P}+", " ", uni)
+    return re.sub(r"\p{P}+", " ", uni)
 
 
 def strip_whitespace(uni):
@@ -22,7 +22,7 @@ def strip_whitespace(uni):
     Returns:
         unicode
     """
-    return re.sub(ur"\s+", u"", uni)
+    return re.sub(r"\s+", "", uni)
 
 
 def find_sublist(l, sublist):
@@ -36,7 +36,7 @@ def find_sublist(l, sublist):
     Returns
         int
     """
-    for i in xrange(len(l)):
+    for i in range(len(l)):
         # Check index 0 first for optimization
         if l[i] == sublist[0] and l[i: i + len(sublist)] == sublist:
             return i
@@ -59,7 +59,7 @@ def __init__(self, text):
         Args:
             text (str or unicode)
         """
-        self._text = unicode(text)
+        self._text = str(text)
         self._tokens = None
 
     @property
diff --git a/python/requirements.txt b/python/requirements.txt
index 301e912e..17cad0c7 100644
--- a/python/requirements.txt
+++ b/python/requirements.txt
@@ -1,3 +1,4 @@
-Pillow>=4.0
-selenium>=3.4.3
+Pillow
+selenium>=4.5.0
 pytest
+numpy
diff --git a/python/test/test_action.py b/python/test/test_action.py
index e2f65491..d6bed732 100644
--- a/python/test/test_action.py
+++ b/python/test/test_action.py
@@ -22,7 +22,7 @@ class RepeatedTester(object):
     def env(self):
         env = MiniWoBEnvironment(self.TASK_NAME)
         base_url = os.environ.get('MINIWOB_BASE_URL')
-        print 'BASE URL:', base_url
+        print('BASE URL:', base_url)
         if self.FRAGILE is True:
             env.configure(base_url=base_url,
                     num_instances=1, seeds=[1], wait_ms=300)
@@ -31,21 +31,21 @@ def env(self):
                     num_instances=1, seeds=[1])
         elif self.FRAGILE == 'delay':
             env.configure(base_url=base_url,
-                    num_instances=3, seeds=range(3), wait_ms=1000)
+                    num_instances=3, seeds=list(range(3)), wait_ms=1000)
         else:
             env.configure(base_url=base_url,
-                    num_instances=3, seeds=range(3))
+                    num_instances=3, seeds=list(range(3)))
         yield env
         env.close()
 
     def test_run(self, env):
-        for i in xrange(self.N):
-            print 'Iteration {} / {}'.format(i + 1, self.N)
+        for i in range(self.N):
+            print('Iteration {} / {}'.format(i + 1, self.N))
             states = env.reset()
             for j, state in enumerate(states):
-                print 'Fields {}: {}'.format(j, state.fields)
-            for s in xrange(self.MAX_STEPS):
-                print 'Step {} / {}'.format(s + 1, self.MAX_STEPS)
+                print('Fields {}: {}'.format(j, state.fields))
+            for s in range(self.MAX_STEPS):
+                print('Step {} / {}'.format(s + 1, self.MAX_STEPS))
                 actions = [self.get_action(x, s) for x in states]
                 states, rewards, dones, info = env.step(actions)
                 if all(x for x in dones):
@@ -61,7 +61,7 @@ def get_action(self, state, step):
 
     def create_element_click_action(self, element):
         action = MiniWoBElementClick(element, fail_hard=True)
-        print 'Clicking with {}'.format(action)
+        print('Clicking with {}'.format(action))
         return action
 
     def click_button(self, state, text):
@@ -75,17 +75,17 @@ def create_coord_click_action(self, element):
         action = MiniWoBCoordClick(
                 element.left + (element.width / 2),
                 element.top + (element.height / 2))
-        print 'Clicking with {}'.format(action)
+        print('Clicking with {}'.format(action))
         return action
 
     def create_type_action(self, text):
         action = MiniWoBType(text)
-        print 'Typing with {}'.format(action)
+        print('Typing with {}'.format(action))
         return action
 
     def create_focus_and_type_action(self, element, text):
         action = MiniWoBFocusAndType(element, text)
-        print 'Focus and Type: {}'.format(action)
+        print('Focus and Type: {}'.format(action))
         return action
 
 
@@ -141,7 +141,7 @@ def _identify_shape(self, state):
         for element in state.dom_elements:
             if element.tag == 'svg':
                 child = element.children[0]
-                print child
+                print(child)
                 if child.tag == 'circle':
                     return 'Circle'
                 elif child.tag == 'text':
@@ -196,7 +196,7 @@ def get_action(self, state, step):
             target = state.fields['target']
             if len(target) > 2:
                 # Hmm... Let's try the LEFT arrow key
-                target = target[:-2] + target[-1] + u'\ue012' + target[-2]
+                target = target[:-2] + target[-1] + '\ue012' + target[-2]
             return self.create_type_action(target)
         elif step == 2:
             # Click submit
@@ -399,7 +399,7 @@ class TestClickPie(RepeatedTester):
     MAX_STEPS = 2
 
     def get_action(self, state, step):
-        print state.dom.visualize()
+        print(state.dom.visualize())
         if step == 0:
             path = None
             for element in state.dom_elements:
@@ -409,7 +409,7 @@ def get_action(self, state, step):
                     return self.create_element_click_action(path)
             assert False, 'Select not found'
         elif step == 1:
-            print state.dom.visualize()
+            print(state.dom.visualize())
             path = None
             for element in state.dom_elements:
                 if element.tag == 'path':
diff --git a/python/test/test_environment.py b/python/test/test_environment.py
index fa8a4151..f84d0ff8 100644
--- a/python/test/test_environment.py
+++ b/python/test/test_environment.py
@@ -16,7 +16,7 @@ class MiniWoBTester(object):
     def env(self):
         env = MiniWoBEnvironment(self.TASK_NAME)
         base_url = os.environ.get('MINIWOB_BASE_URL')
-        print 'BASE URL:', base_url
+        print('BASE URL:', base_url)
         env.configure(num_instances=3, seeds=[1,2,'hello'], base_url=base_url)
         yield env
         env.close()
@@ -33,7 +33,7 @@ def get_coord_click(self, state):
         for element in state.dom_elements:
             if element.tag == 'button':
                 action = MiniWoBCoordClick(element.left + 5, element.top + 5)
-                print 'Clicking with {}'.format(action)
+                print('Clicking with {}'.format(action))
                 return action
         raise ValueError('Cannot find button: {}'.format(str(state.dom_elements)))
 
@@ -42,7 +42,7 @@ def get_element_click(self, state):
         for element in state.dom_elements:
             if element.tag == 'button':
                 action = MiniWoBElementClick(element, fail_hard=True)
-                print 'Clicking with {}'.format(action)
+                print('Clicking with {}'.format(action))
                 return action
         raise ValueError('Cannot find button: {}'.format(str(state.dom_elements)))
 
@@ -58,110 +58,110 @@ def test_do_nothing(self, env):
 
     def test_run(self, env):
         """Test reset and step."""
-        print '=' * 40
+        print('=' * 40)
         states = env.reset()
-        print [x.utterance for x in states]
+        print([x.utterance for x in states])
         assert all(x.utterance == 'Click the button.' for x in states)
-        print [x.fields for x in states]
+        print([x.fields for x in states])
         assert all(x.fields.keys == ['dummy'] for x in states)
-        print [x.dom for x in states]
-        print states[0].dom_elements
-        print states[0].dom.visualize()
+        print([x.dom for x in states])
+        print(states[0].dom_elements)
+        print(states[0].dom.visualize())
         ################################
-        print '=' * 40
+        print('=' * 40)
         states, rewards, dones, info = env.step([None, None, None])
-        print [x.utterance for x in states]
-        print [x.dom for x in states]
-        print rewards
+        print([x.utterance for x in states])
+        print([x.dom for x in states])
+        print(rewards)
         assert rewards == [0, 0, 0]
-        print dones
+        print(dones)
         assert dones == [False, False, False]
-        print info
+        print(info)
         ################################
         # Test clicking
-        print '=' * 40
+        print('=' * 40)
         action = self.get_click(states[1])
         states, rewards, dones, info = env.step([None, action, None])
         assert states[1] is None
         states = [states[0], states[2]]
-        print [x.utterance for x in states]
-        print [x.dom for x in states]
-        print rewards
+        print([x.utterance for x in states])
+        print([x.dom for x in states])
+        print(rewards)
         assert rewards[0] == 0 and rewards[1] > 0 and rewards[2] == 0
-        print dones
+        print(dones)
         assert dones == [False, True, False]
-        print info
+        print(info)
         ################################
         # Wait for timeout
-        print '=' * 40
-        for i in xrange(1, 13):
-            print 'Sleeping ... ({})'.format(i)
+        print('=' * 40)
+        for i in range(1, 13):
+            print('Sleeping ... ({})'.format(i))
             time.sleep(1)
         states, rewards, dones, info = env.step([None, None, None])
         assert states == [None, None, None]
-        print rewards
+        print(rewards)
         assert rewards[0] < 0 and rewards[1] > 0 and rewards[2] < 0
-        print dones
+        print(dones)
         assert dones == [True, True, True]
-        print info
+        print(info)
         ################################
         # Start again
-        print '=' * 40
+        print('=' * 40)
         states = env.reset()
-        print [x.utterance for x in states]
-        print [x.dom for x in states]
+        print([x.utterance for x in states])
+        print([x.dom for x in states])
         ################################
-        print '=' * 40
+        print('=' * 40)
         states, rewards, dones, info = env.step([None, None, None])
-        print [x.utterance for x in states]
-        print [x.dom for x in states]
-        print rewards
+        print([x.utterance for x in states])
+        print([x.dom for x in states])
+        print(rewards)
         assert rewards == [0, 0, 0]
-        print dones
+        print(dones)
         assert dones == [False, False, False]
-        print info
+        print(info)
 
     def test_speed(self, env):
         states = env.reset()
         start_time = time.time()
         elapsed = []
         N = 50
-        for i in xrange(1, N + 1):
-            print 'Iteration', i, '/', N
+        for i in range(1, N + 1):
+            print('Iteration', i, '/', N)
             actions = [None] * len(states)
             states, rewards, dones, info = env.step(actions)
             elapsed.append(time.time() - start_time)
             start_time = time.time()
         mean = sum(elapsed) / len(elapsed)
-        print 'Average time:', mean
-        print 'SD:', sum((x - mean) ** 2 for x in elapsed) / len(elapsed)
+        print('Average time:', mean)
+        print('SD:', sum((x - mean) ** 2 for x in elapsed) / len(elapsed))
 
     def test_reset(self, env):
-        print '=' * 40
+        print('=' * 40)
         states = env.reset()
         # Test clicking
         action = self.get_click(states[1])
         states, rewards, dones, info = env.step([None, action, None])
         assert dones == [False, True, False]
-        print info
+        print(info)
         # Should issue a warning on the second instance
         states, rewards, dones, info = env.step([None, action, None])
         assert dones == [False, True, False]
-        print info
+        print(info)
         # Hard reset
         states = env.reset()
         states, rewards, dones, info = env.step([None, None, None])
         assert dones == [False, False, False]
-        print info
+        print(info)
         # Now click the first and second
         action_0 = self.get_click(states[0])
         action_1 = self.get_click(states[1])
         states, rewards, dones, info = env.step([action_0, action_1, None])
         assert dones == [True, True, False]
-        print info
+        print(info)
 
     def test_attention(self, env):
-        print '=' * 40
+        print('=' * 40)
         states = env.reset()
         attention = np.random.rand(20, 20) * .02
         env.visualize_attention([attention, None, None])
@@ -170,7 +170,7 @@ def test_attention(self, env):
         time.sleep(2)
 
     def test_suicide(self, env):
-        print '=' * 40
+        print('=' * 40)
         states = env.reset()
         action = MiniWoBTerminate()
         states, rewards, dones, info = env.step([None, action, None])
@@ -188,7 +188,7 @@ def get_button_click(self, state):
         for element in state.dom_elements:
             if element.tag == 'button' and element.text == state.fields['target']:
                 action = MiniWoBElementClick(element, fail_hard=True)
-                print 'Clicking with {}'.format(action)
+                print('Clicking with {}'.format(action))
                 return action
         raise ValueError('Cannot find button: {}'.format(str(state.dom_elements)))
 
@@ -197,14 +197,14 @@ def get_input_click(self, state):
         for element in state.dom_elements:
             if element.tag == 'input_text':
                 action = MiniWoBElementClick(element, fail_hard=True)
-                print 'Clicking with {}'.format(action)
+                print('Clicking with {}'.format(action))
                 return action
         raise ValueError('Cannot find input: {}'.format(str(state.dom_elements)))
 
     def test_seed(self, env):
-        print '=' * 40
+        print('=' * 40)
         states = env.reset(seeds=['hello', 'itsme', 'hello'])
-        print states[0].dom.visualize()
+        print(states[0].dom.visualize())
         # Check that everything is the same for instances 0 and 2
         utt_hello = states[0].utterance
         utt_itsme = states[1].utterance
@@ -247,7 +247,7 @@ def get_button_click(self, state, text):
         for element in state.dom_elements:
             if element.tag == 'button' and element.text == text:
                 action = MiniWoBElementClick(element, fail_hard=True)
-                print 'Clicking with {}'.format(action)
+                print('Clicking with {}'.format(action))
                 return action
         raise ValueError('Cannot find button: {}'.format(str(state.dom_elements)))
 
@@ -256,7 +256,7 @@ def test_mode(self, env):
         - mode = 'train': click on button ONE
         - mode = 'test':  click on button TWO
         """
-        print '=' * 40
+        print('=' * 40)
         # Training time
         states = env.reset()
         targets = ['ONE', 'TWO', 'ONE']
@@ -303,12 +303,12 @@ class TestMiniWoBFields(MiniWoBTester):
     TASK_NAME = 'email-inbox-forward-nl'
 
     def test_fields(self, env):
-        print '=' * 40
+        print('=' * 40)
         # Training time
         states = env.reset()
         for state in states:
-            print state.utterance
-            print state.fields
+            print(state.utterance)
+            print(state.fields)
             assert 'by' in state.fields.keys
             assert 'to' in state.fields.keys
             assert state.fields['by'] in state.utterance
@@ -316,15 +316,15 @@ def test_fields(self, env):
         # Test time
         states = env.reset(mode='test')
         for state in states:
-            print state.utterance
-            print state.fields
+            print(state.utterance)
+            print(state.fields)
             assert state.fields.keys == ['dummy']
             assert state.utterance
         # Training time
         states = env.reset(mode='train')
         for state in states:
-            print state.utterance
-            print state.fields
+            print(state.utterance)
+            print(state.fields)
             assert 'by' in state.fields.keys
             assert 'to' in state.fields.keys
             assert state.fields['by'] in state.utterance
diff --git a/record.py b/record.py
index 46362b30..d5328d93 100755
--- a/record.py
+++ b/record.py
@@ -7,7 +7,7 @@
 
 import sys, os, shutil, re, argparse, json, time
 from codecs import open
-from itertools import izip
+
 from collections import defaultdict, Counter
 
 from bottle import Bottle, request, response
@@ -33,7 +33,7 @@ def save(self, data):
             filename += 'x'
         with open(filename, 'w') as fout:
             json.dump(data, fout)
-        print 'Saved to {}'.format(filename)
+        print('Saved to {}'.format(filename))
         return filename
 
     def save_turk(self, request):
@@ -49,7 +49,7 @@ def save_turk(self, request):
                 filename += 'x'
             with open(filename, 'w') as fout:
                 fout.write(data)
-            print 'Saved to {}'.format(filename)
+            print('Saved to {}'.format(filename))
 
     @staticmethod
     def decompress_turk(compressed):
@@ -110,7 +110,7 @@ def main():
     # Start the server
     host = 'localhost' if not args.global_access else '0.0.0.0'
     app.run(host=host, port=args.port)
-    print '\nGood bye!'
+    print('\nGood bye!')
     
 
 if __name__ == '__main__':

From f2a1f23c21416c822184c9ba6b1d241367aaf171 Mon Sep 17 00:00:00 2001
From: Panupong Pasupat <ppasupat@stanford.edu>
Date: Sat, 22 Oct 2022 23:38:01 -0700
Subject: [PATCH 2/2] Minor Python nit

---
 python/miniwob/fields.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/miniwob/fields.py b/python/miniwob/fields.py
index c29d8790..e668473d 100644
--- a/python/miniwob/fields.py
+++ b/python/miniwob/fields.py
@@ -52,7 +52,7 @@ def values(self):
         return list(self._d.values())
 
     def __repr__(self):
-        return '\n'.join('{}: {}'.format(k, repr(v)) for k, v in list(self._d.items()))
+        return '\n'.join('{}: {}'.format(k, repr(v)) for k, v in self._d.items())
     __str__ = __repr__