def _make_env_task()

in gym_genesis/env.py [0:0]

• 22 lines of code
• 3 McCabe index (conditional complexity)

    def _make_env_task(self, task_name):
        if task_name == "cube_pick":
            task = CubePick(enable_pixels=self.enable_pixels,
                            observation_height=self.observation_height, 
                            observation_width=self.observation_width,
                            num_envs = self.num_envs,
                            env_spacing = self.env_spacing,
                            camera_capture_mode = self.camera_capture_mode,
                            strip_environment_state=self.strip_environment_state,
                            )
        elif task_name == "cube_stack":
            task = CubeStack(enable_pixels=self.enable_pixels,
                            observation_height=self.observation_height, 
                            observation_width=self.observation_width,
                            num_envs = self.num_envs,
                            env_spacing = self.env_spacing,
                            camera_capture_mode = self.camera_capture_mode,
                            strip_environment_state=self.strip_environment_state,
                            )
        else:
            raise NotImplementedError(task_name)
        return task