def reset()

in gym_wikinav/envs/wikinav_env/web_graph.py [0:0]

• 10 lines of code
• 1 McCabe index (conditional complexity)

    def reset(self, is_training=True):
        """
        Prepare a new navigation rollout.
        """
        # TODO: Sample outside of the training set.
        ids, paths, lengths = self.graph.sample_paths(1, is_training)
        self._id, self._path, self._length = ids[0], paths[0], lengths[0]
        self._cur_article_id = self._path[0]

        self._target_id = self._path[self._length - 2]
        self._on_target = False
        self._success, self._stopped = False, False

        self._num_steps = 0
        self._reset(is_training)
        self._prepare()