def __init__()

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

• 12 lines of code
• 2 McCabe index (conditional complexity)

    def __init__(self, beam_size=32, graph=None, goal_reward=10.0):
        """
        Args:
            beam_size: Number of candidates to present as actions at each
                timestep
            graph:
        """
        super(WikiNavEnv, self).__init__()

        if graph is None:
            graph = web_graph.EmbeddedWikispeediaGraph.get_default_graph()
        self.graph = graph

        # TODO verify beam size

        self.beam_size = beam_size
        self.goal_reward = goal_reward

        self.path_length = self.graph.path_length

        self.navigator = web_graph.Navigator(self.graph, self.beam_size,
                                             self.path_length)

        self._action_space = spaces.Discrete(self.beam_size)

        self._just_reset = False